debug

app.py

@@ -15,6 +15,7 @@ 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:
@@ -32,23 +33,15 @@ 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")
-
-        # 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}")
+        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
+        )
 
     def respond(self, message, history):
         messages = [{"role": "system", "content": SYSTEM_PROMPT}]
@@ -72,38 +65,14 @@ class SessionChatBot:
                 response += token
                 yield response
 
-        # Save after generation completes
+        # Save log after full response
         self.append_to_session_log(message, response)
 
-# ---- Gradio Interface ----
-with gr.Blocks() as demo:
-    chatbot_state = gr.State()
-
-    chatbot_ui = gr.Chatbot()
-    msg_input = gr.Textbox(label="Message", placeholder="Say something...")
-    send_btn = gr.Button("Send")
-
-    # Initialize a new bot per user
-    def init_bot():
-        return SessionChatBot()
-
-    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_input, chatbot_ui, chatbot_state],
-        outputs=[chatbot_ui]
-    )
+# ---- Instantiate a new bot for each user session ----
+def create_chatbot():
+    return SessionChatBot().respond
 
-    msg_input.submit(
-        handle_message,
-        inputs=[msg_input, chatbot_ui, chatbot_state],
-        outputs=[chatbot_ui]
-    )
+demo = gr.ChatInterface(fn=create_chatbot(), title="BoundrAI")
 
-# Launch app
 if __name__ == "__main__":
     demo.launch()