back to simple session ID

app.py

@@ -45,12 +45,14 @@ class SessionChatBot:
 
     def respond(self, message, history):
         messages = [{"role": "system", "content": SYSTEM_PROMPT}]
-        for msg in history:
-            messages.append(msg)
+        for user_msg, bot_msg in history:
+            if user_msg:
+                messages.append({"role": "user", "content": user_msg})
+            if bot_msg:
+                messages.append({"role": "assistant", "content": bot_msg})
         messages.append({"role": "user", "content": message})
 
         response = ""
-        # Yield a full message dict each time (not just a string)
         for chunk in client.chat_completion(
             messages,
             max_tokens=512,
@@ -61,9 +63,9 @@ class SessionChatBot:
             token = chunk.choices[0].delta.content
             if token:
                 response += token
-                yield {"role": "assistant", "content": response}
+                yield response
 
-        # Save full response to log
+        # Save log after full response
         self.append_to_session_log(message, response)
 
     def report_interaction(self):
@@ -76,13 +78,16 @@ class SessionChatBot:
         if not lines:
             return "No conversation to report."
 
+        # Mark last interaction as reported
         last_entry = json.loads(lines[-1])
         last_entry["reported"] = True
         lines[-1] = json.dumps(last_entry) + "\n"
 
+        # Overwrite file
         with open(self.local_log_path, "w", encoding="utf-8") as f:
             f.writelines(lines)
 
+        # Upload updated log to Hugging Face
         api.upload_file(
             path_or_fileobj=self.local_log_path,
             path_in_repo=self.remote_log_path,
@@ -92,28 +97,22 @@ class SessionChatBot:
         )
         return "Interaction reported successfully."
 
-# ---- Build Gradio Interface with per-user chatbot ----
-with gr.Blocks() as demo:
-    chatbot_state = gr.State()
+# ---- Instantiate and Share Bot Session Globally ----
+chatbot_instance = SessionChatBot()
 
-    chatbot_ui = gr.ChatInterface(
-        fn=lambda message, history, chatbot: chatbot.respond(message, history),
-        title="BoundrAI",
-        textbox=gr.Textbox(placeholder="Ask me anything...", container=True),
-        additional_inputs=[chatbot_state],
-    )
+def create_chatbot():
+    return chatbot_instance.respond
 
+# ---- Build Gradio Interface ----
+with gr.Blocks() as demo:
+    chatbot = gr.ChatInterface(fn=create_chatbot(), title="BoundrAI")
     report_btn = gr.Button("Report Companion Interaction")
     status_box = gr.Textbox(label="Report Status", interactive=False)
 
-    def setup_chatbot():
-        return SessionChatBot()
-
-    def report(chatbot):
-        return chatbot.report_interaction()
+    def report():
+        return chatbot_instance.report_interaction()
 
-    demo.load(fn=setup_chatbot, outputs=chatbot_state)
-    report_btn.click(fn=report, inputs=[chatbot_state], outputs=status_box)
+    report_btn.click(fn=report, outputs=status_box)
 
 if __name__ == "__main__":
     demo.launch()