Update app.py

app.py

@@ -158,13 +158,23 @@ def generate_response(user_input, history, mode="chat"):
 # ---------------------------
 # 7. Gradio Chat UI
 # ---------------------------
+
+
+# ---------------------------
+# 7. Chat Logic
+# ---------------------------
 def chat_with_model(user_message, chat_history):
     if not user_message:
         return chat_history, ""
 
-    # Maintain chat history for coherence
-    history = [{"role": "assistant" if i % 2 else "user", "content": msg}
-               for i, (msg, _) in enumerate(chat_history)]
+    # Convert Gradio's new message format to your history list
+    history = [
+        {"role": m["role"], "content": m["content"]}
+        for m in chat_history
+        if isinstance(m, dict) and "role" in m
+    ]
+
+    history.append({"role": "user", "content": user_message})
 
     try:
         bot_text = generate_response(user_message, history)
@@ -172,14 +182,17 @@ def chat_with_model(user_message, chat_history):
         tb = traceback.format_exc()
         bot_text = f"⚠️ Error: {e}\n\n{tb}"
 
-    chat_history.append((user_message, bot_text))
+    chat_history.append({"role": "assistant", "content": bot_text})
     return chat_history, ""
 
+
 def reset_chat():
     return []
 
-import gradio as gr
 
+# ---------------------------
+# 8. Gradio Chat UI
+# ---------------------------
 def build_ui():
     with gr.Blocks(
         theme=gr.themes.Soft(primary_hue="indigo"),
@@ -225,17 +238,27 @@ def build_ui():
         """
     ) as demo:
 
+        # Header
         gr.Markdown(
             """
             <div style="text-align:center; margin-bottom: 1rem;">
                 <h2 style="font-weight:700; font-size:1.8rem;">💠 OhamLab AI Assistant</h2>
-                <p style="color:#6b7280; font-size:0.95rem;">Your intelligent R&D and engineering companion powered by contextual knowledge.</p>
+                <p style="color:#6b7280; font-size:0.95rem;">
+                    Your intelligent R&D and engineering companion — powered by OhamLab contextual knowledge.
+                </p>
             </div>
             """
         )
 
-        chatbot = gr.Chatbot(height=520, elem_id="ohamlab", type="tuples")
+        # Chatbot (new format)
+        chatbot = gr.Chatbot(
+            height=520,
+            elem_id="ohamlab",
+            type="messages",  # ✅ modern format
+            avatar_images=("🧑‍💻", "🤖"),
+        )
 
+        # Input box (full width)
         with gr.Row():
             msg = gr.Textbox(
                 placeholder="Type your message here...",
@@ -245,20 +268,23 @@ def build_ui():
                 container=False,
             )
 
-        with gr.Row(equal_height=True):
+        # Buttons (Send + Clear aligned below)
+        with gr.Row(equal_height=True, variant="compact"):
             send = gr.Button("Send", variant="primary", elem_classes=["primary"])
             clear = gr.Button("Clear", variant="secondary", elem_classes=["secondary"])
 
-        # Wire events
+        # Wiring
         send.click(chat_with_model, inputs=[msg, chatbot], outputs=[chatbot, msg])
         msg.submit(chat_with_model, inputs=[msg, chatbot], outputs=[chatbot, msg])
         clear.click(reset_chat, outputs=chatbot)
 
     return demo
 
+
 # ---------------------------
 # Entrypoint
 # ---------------------------
 if __name__ == "__main__":
-    print("🚀 Starting OhamLab ")
-    build_ui()
+    print("🚀 Starting OhamLab Assistant...")
+    demo = build_ui()
+    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)