clean up interface

app.py

@@ -45,8 +45,11 @@ 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 = ""
@@ -60,7 +63,7 @@ class SessionChatBot:
             token = chunk.choices[0].delta.content
             if token:
                 response += token
-                yield {"role": "assistant", "content": response}
+                yield response
 
         # Save log after full response
         self.append_to_session_log(message, response)
@@ -102,21 +105,14 @@ def create_chatbot():
 
 # ---- Build Gradio Interface ----
 with gr.Blocks() as demo:
-    with gr.Column(scale=1):
-        chatbot_ui = gr.ChatInterface(
-            fn=create_chatbot(),
-            title="BoundrAI",
-            type="messages",  # Switch to OpenAI-style message format
-            textbox=gr.Textbox(placeholder="Ask me anything...", container=True),
-        )
-
-        report_btn = gr.Button("Report Companion Interaction")
-        status_box = gr.Textbox(label="Report Status", interactive=False)
+    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 report():
-            return chatbot_instance.report_interaction()
+    def report():
+        return chatbot_instance.report_interaction()
 
-        report_btn.click(fn=report, outputs=status_box)
+    report_btn.click(fn=report, outputs=status_box)
 
 if __name__ == "__main__":
     demo.launch()