Upload app.py

app.py

@@ -522,63 +522,60 @@ if __name__ == "__main__":
     server_name = os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0")
     server_port = int(os.environ.get("GRADIO_SERVER_PORT", 7860))
     
-    # Create a SIMPLE Gradio UI that works with any version
-    with gr.Blocks(title="Codey Bryant 3.0 - SOTA RAG Coding Assistant") as demo:
-        gr.Markdown("""
-        # 🤖 Codey Bryant 3.0
-        ## **SOTA RAG Coding Assistant**
+    # SIMPLE, WORKING UI
+    with gr.Blocks() as demo:
+        gr.Markdown("# 🤖 Codey Bryant 3.0")
+        gr.Markdown("Python Coding Assistant with SOTA RAG")
         
-        **Features:** HyDE + Query Rewriting + Multi-Query + Answer-Space Retrieval
+        # Status and initialization
+        status = gr.Textbox("Click 'Initialize' to start", label="Status", interactive=False)
+        init_btn = gr.Button("Initialize", variant="primary")
         
-        **Handles vague questions:** "it's not working", "help with error", "make it faster"
-        """)
+        # Chat interface
+        chatbot = gr.Chatbot(label="Conversation")
+        msg = gr.Textbox(placeholder="Type your Python question here...", label="Your Question")
+        submit = gr.Button("Send")
+        clear = gr.Button("Clear")
         
-        init_btn = gr.Button("Initialize Assistant", variant="primary")
-        status = gr.Markdown("**Status:** Click 'Initialize Assistant' to start")
-        
-        chatbot = gr.Chatbot(height=500)
-        msg = gr.Textbox(
-            placeholder="Ask anything about Python coding...",
-            label="Your Question",
-            lines=3
-        )
-        clear = gr.Button("Clear Chat")
+        # Event handlers
+        def init_and_enable():
+            yield from initialize_assistant()
+            return gr.update(interactive=True)
         
-        # Examples
-        gr.Examples(
-            examples=[
-                ["How to read a CSV file in Python?"],
-                ["Why am I getting 'list index out of range' error?"],
-                ["Help, my code isn't working!"],
-            ],
-            inputs=msg
+        init_btn.click(
+            init_and_enable,
+            outputs=[status]
         )
         
-        # Event handlers
-        init_btn.click(initialize_assistant, outputs=status)
+        def add_message(message, history):
+            return "", history + [[message, None]]
         
         def respond(message, history):
-            history.append([message, ""])
-            return "", history
+            for response in chat(message, history):
+                yield response
         
-        msg.submit(
+        # Connect the send button
+        submit.click(
+            add_message,
+            [msg, chatbot],
+            [msg, chatbot]
+        ).then(
             respond,
             [msg, chatbot],
+            chatbot
+        )
+        
+        # Also connect Enter key
+        msg.submit(
+            add_message,
+            [msg, chatbot],
             [msg, chatbot]
         ).then(
-            chat,
+            respond,
             [msg, chatbot],
             chatbot
         )
         
         clear.click(lambda: [], None, chatbot)
     
-    # Launch the app
-    logger.info(f"Starting Codey Bryant 3.0 on {server_name}:{server_port}")
-    logger.info("SOTA RAG Architecture: HyDE + Query Rewriting + Multi-Query + Answer-Space Retrieval")
-    
-    demo.launch(
-        server_name=server_name,
-        server_port=server_port,
-        share=False
-    )
+    demo.launch(server_name=server_name, server_port=server_port, share=False)