Update ui.py

ui.py

@@ -1,66 +1,37 @@
-# ui.py - FINAL FIXED VERSION
 import gradio as gr
-import asyncio
 
 def create_ui(generate_fn):
     """
-    Build a Blocks-based chat UI.
-    Correctly handles async generators without SyntaxError.
+    Build a Blocks-based chat UI that returns a gr.Blocks object.
+    generate_fn receives a list of messages (user/assistant pairs as dicts)
+    and should return a reply string (this wrapper uses it as a placeholder).
     """
     with gr.Blocks(css="""
         body { background-color: #0b0f19; color: #e0e0e0; }
         .gradio-container { font-family: 'Inter', sans-serif; max-width: 900px !important; margin: auto; }
         footer { visibility: hidden; display: none; }
-    """, title="Nexari AI") as demo:
-        
+    """) as demo:
         gr.Markdown("## Nexari AI — Research Backend")
-        
-        # FIX: Removed .style() (deprecated) and used height parameter
-        chatbot = gr.Chatbot(
-            elem_id="nexari_chatbot", 
-            label="Nexari G1", 
-            height=600,
-            bubble_full_width=False
-        )
-        
-        txt = gr.Textbox(
-            show_label=False, 
-            placeholder="Type your message and press Enter...",
-            container=False
-        )
-        
-        # State to hold conversation history
+        chatbot = gr.Chatbot(elem_id="nexari_chatbot", label="Nexari").style(height=500)
+        txt = gr.Textbox(show_label=False, placeholder="Type your message and press Enter")
         state = gr.State([])
 
-        async def submit(message, history):
-            if not message:
-                yield history, ""
-                return
-
-            # Prepare messages list for the backend
+        def submit(message, history):
+            # history is list of (human, ai) pairs in Gradio; convert to messages
             messages = []
             for human, ai in history:
                 messages.append({"role": "user", "content": human})
                 messages.append({"role": "assistant", "content": ai})
             messages.append({"role": "user", "content": message})
-            
-            # Add user message to UI immediately
-            history.append((message, ""))
-            yield history, ""
-
-            # Consume the generator from app.py
-            # FIX: Used async for loop correctly without illegal return
-            partial_response = ""
-            async for chunk in generate_fn(messages):
-                partial_response = chunk
-                history[-1] = (message, partial_response)
-                yield history, ""
-            
-            # Generator ends naturally here. No return statement needed with value.
+            # call generator wrapper (synchronous placeholder)
+            reply = generate_fn(messages)
+            # append to history
+            history = history + [(message, reply)]
+            return history, ""
 
-        # Bind events
-        txt.submit(submit, [txt, chatbot], [chatbot, txt], queue=True)
+        txt.submit(submit, [txt, state], [chatbot, txt], queue=False)
+        # Also allow clicking a send button
         send = gr.Button("Send")
-        send.click(submit, [txt, chatbot], [chatbot, txt], queue=True)
-        
+        send.click(submit, [txt, state], [chatbot, txt], queue=False)
+
     return demo