Upload 2 files

app.py

@@ -476,149 +476,109 @@ def chat(message: str, history: list):
         history[-1][1] = f"Error: {str(e)}"
         yield history
 
-def create_ui():
-    """Create Gradio UI optimized for Hugging Face"""
-    # Use a compatible theme
-    theme = gr.themes.Soft()
+# 4) Gradio UI
+
+ASSISTANT: Optional[HybridCodeAssistant] = None
+
+def initialize_assistant():
+    """Initialize assistant with progress tracking"""
+    global ASSISTANT
+    if ASSISTANT is None:
+        yield "Initializing Codey Bryant 3.0..."
+        yield "Loading retrieval system..."
+        ASSISTANT = HybridCodeAssistant()
+        yield "Codey Bryant 3.0 Ready!"
+        yield "SOTA RAG Features: HyDE + Query Rewriting + Multi-Query + Answer-Space Retrieval"
+        yield "Ask coding questions like: 'it's not working', 'help with error', 'make it faster'"
+    else:
+        yield "Assistant already initialized!"
+
+def chat(message: str, history: list):
+    """Chat function with error handling"""
+    if ASSISTANT is None:
+        yield "Please click 'Initialize Assistant' first!"
+        return
+
+    # Append user message
+    history.append([message, ""])
+    yield history
+
+    # Stream response
+    try:
+        response = ""
+        for token in ASSISTANT.answer_stream(message):
+            response += token
+            history[-1][1] = response
+            yield history
+    except Exception as e:
+        logger.error(f"Chat error: {e}")
+        history[-1][1] = f"Error: {str(e)}"
+        yield history
+
+# 5) Main Entry Point
+
+if __name__ == "__main__":
+    # Configure for Hugging Face Spaces
+    server_name = os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0")
+    server_port = int(os.environ.get("GRADIO_SERVER_PORT", 7860))
     
-    with gr.Blocks(
-        title="Codey Bryant 3.0 - SOTA RAG Coding Assistant",
-        theme=theme,  # Pass theme here
-        css="""
-        .gradio-container { max-width: 1200px; margin: auto; }
-        .chatbot { min-height: 500px; }
-        .status-box { padding: 20px; border-radius: 10px; background: #f0f8ff; }
-        """
-    ) as demo:
+    # 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**
         
-        ### **Advanced Features:**
-        - **HyDE** (Hypothetical Document Embeddings)
-        - **Query Rewriting** for vague queries
-        - **Multi-Query** retrieval
-        - **Answer-Space Retrieval**
+        **Features:** HyDE + Query Rewriting + Multi-Query + Answer-Space Retrieval
         
-        ### **Handles vague questions like:**
-        - "it's not working" 
-        - "help with error"
-        - "make it faster" 
-        - "why error"
-        - "how to implement"
-        
-        ### **Powered by:**
-        - TinyLlama 1.1B (4-bit quantized)
-        - Hybrid retrieval (FAISS + BM25)
-        - OPC coding datasets
+        **Handles vague questions:** "it's not working", "help with error", "make it faster"
         """)
         
-        with gr.Row():
-            with gr.Column(scale=1):
-                init_btn = gr.Button(
-                    "Initialize Assistant",
-                    variant="primary",
-                    size="lg"
-                )
-                clear_btn = gr.Button("Clear Chat", size="lg")
-                
-            with gr.Column(scale=4):
-                status = gr.Markdown(
-                    "### Status: Click 'Initialize Assistant' to start",
-                    elem_classes="status-box"
-                )
+        init_btn = gr.Button("Initialize Assistant", variant="primary")
+        status = gr.Markdown("**Status:** Click 'Initialize Assistant' to start")
         
-        chatbot = gr.Chatbot(
-            label="Chat with Codey",
-            height=500,
-            show_label=True,
-            avatar_images=(None, "🤖"),
-            bubble_full_width=False
+        chatbot = gr.Chatbot(height=500)
+        msg = gr.Textbox(
+            placeholder="Ask anything about Python coding...",
+            label="Your Question",
+            lines=3
         )
-        
-        with gr.Row():
-            msg = gr.Textbox(
-                placeholder="Ask anything about Python coding...",
-                label="Your Question",
-                lines=3,
-                scale=5,
-                container=False
-            )
-            submit_btn = gr.Button("Send", variant="secondary", scale=1)
+        clear = gr.Button("Clear Chat")
         
         # Examples
         gr.Examples(
             examples=[
                 ["How to read a CSV file in Python?"],
                 ["Why am I getting 'list index out of range' error?"],
-                ["Make this function faster..."],
                 ["Help, my code isn't working!"],
-                ["Best way to sort a dictionary by value?"]
             ],
-            inputs=msg,
-            label="Try these examples:"
+            inputs=msg
         )
         
         # Event handlers
-        init_btn.click(
-            initialize_assistant,
-            outputs=status
-        )
+        init_btn.click(initialize_assistant, outputs=status)
         
-        def submit_message(message, history):
-            return "", history + [[message, None]]
+        def respond(message, history):
+            history.append([message, ""])
+            return "", history
         
         msg.submit(
-            submit_message,
+            respond,
             [msg, chatbot],
-            [msg, chatbot],
-            queue=False
+            [msg, chatbot]
         ).then(
             chat,
             [msg, chatbot],
             chatbot
         )
         
-        submit_btn.click(
-            submit_message,
-            [msg, chatbot],
-            [msg, chatbot],
-            queue=False
-        ).then(
-            chat,
-            [msg, chatbot],
-            chatbot
-        )
-        
-        clear_btn.click(lambda: None, None, chatbot, queue=False)
-        
-        # Footer
-        gr.Markdown("""
-        ---
-        *Codey Bryant 3.0 uses TinyLlama 1.1B with 4-bit quantization. Responses may take a few seconds.*
-        """)
-    
-    return demo
-
-# ========================
-# 5) Main Entry Point
-# ========================
-
-if __name__ == "__main__":
-    # Configure for Hugging Face Spaces
-    server_name = os.environ.get("GRADIO_SERVER_NAME", "0.0.0.0")
-    server_port = int(os.environ.get("GRADIO_SERVER_PORT", 7860))
-    
-    # Create and launch the demo
-    demo = create_ui()
+        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,  # Set to True if you want a public link
-        debug=False,
-        show_error=True
+        share=False
     )