debug

app.py

@@ -4,8 +4,6 @@ from datetime import datetime
 import uuid
 import os
 import json
-from huggingface_hub import HfApi
-
 
 # ---- Configuration ----
 MODEL_NAME = "HuggingFaceH4/zephyr-7b-beta"
@@ -17,7 +15,6 @@ with open("system_prompt.txt", "r") as f:
     SYSTEM_PROMPT = f.read()
 
 client = InferenceClient(MODEL_NAME)
-api = HfApi()
 
 # ---- Chatbot Class with per-user session ID ----
 class SessionChatBot:
@@ -35,15 +32,23 @@ class SessionChatBot:
             "system_prompt": SYSTEM_PROMPT,
             "session_id": self.session_id
         }
+
+        # Write to local log
         with open(self.local_log_path, "a", encoding="utf-8") as f:
             f.write(json.dumps(row) + "\n")
-        api.upload_file(
-            path_or_fileobj=self.local_log_path,
-            path_in_repo=self.remote_log_path,
-            repo_id=DATASET_REPO,
-            repo_type="dataset",
-            token=HF_TOKEN
-        )
+
+        # Upload to Hugging Face dataset
+        try:
+            api = HfApi()  # Reinitialize inside callback context (important for Spaces)
+            api.upload_file(
+                path_or_fileobj=self.local_log_path,
+                path_in_repo=self.remote_log_path,
+                repo_id=DATASET_REPO,
+                repo_type="dataset",
+                token=HF_TOKEN
+            )
+        except Exception as e:
+            print(f"[Upload error] {e}")
 
     def respond(self, message, history):
         messages = [{"role": "system", "content": SYSTEM_PROMPT}]
@@ -67,36 +72,38 @@ class SessionChatBot:
                 response += token
                 yield response
 
-        # Save log after full response
+        # Save after generation completes
         self.append_to_session_log(message, response)
 
-# ---- Gradio Interface with per-user session support ----
+# ---- Gradio Interface ----
 with gr.Blocks() as demo:
-    chatbot_state = gr.State()  # each user gets a different SessionChatBot
+    chatbot_state = gr.State()
 
-    chat = gr.Chatbot()
-    msg = gr.Textbox(label="Message", placeholder="Say something...")
+    chatbot_ui = gr.Chatbot()
+    msg_input = gr.Textbox(label="Message", placeholder="Say something...")
     send_btn = gr.Button("Send")
 
-    # Create a chatbot instance per user when interface loads
+    # Initialize a new bot per user
     def init_bot():
         return SessionChatBot()
 
-    demo.load(init_bot, outputs=chatbot_state)
+    demo.load(init_bot, outputs=[chatbot_state])
 
     def handle_message(message, history, bot: SessionChatBot):
         return bot.respond(message, history)
 
     send_btn.click(
         handle_message,
-        inputs=[msg, chat, chatbot_state],
-        outputs=chat
+        inputs=[msg_input, chatbot_ui, chatbot_state],
+        outputs=[chatbot_ui]
     )
 
-    msg.submit(
+    msg_input.submit(
         handle_message,
-        inputs=[msg, chat, chatbot_state],
-        outputs=chat
+        inputs=[msg_input, chatbot_ui, chatbot_state],
+        outputs=[chatbot_ui]
     )
 
-demo.launch()
+# Launch app
+if __name__ == "__main__":
+    demo.launch()