Update app.py

app.py

@@ -158,40 +158,49 @@ def generate_response(user_input, history, mode="chat"):
 # ---------------------------
 # 7. Gradio Chat UI
 # ---------------------------
-
+import traceback
+import gradio as gr
 
 # ---------------------------
-# 7. Chat Logic
+# Chat Logic
 # ---------------------------
 def chat_with_model(user_message, chat_history):
+    """
+    Maintains full conversational context and returns updated chat history.
+    The assistant speaks as 'OhamLab'.
+    """
     if not user_message:
         return chat_history, ""
 
-    # Convert Gradio's new message format to your history list
+    # Maintain persistent history in OpenAI-style structure
     history = [
         {"role": m["role"], "content": m["content"]}
         for m in chat_history
         if isinstance(m, dict) and "role" in m
     ]
 
+    # Append user query
     history.append({"role": "user", "content": user_message})
 
     try:
         bot_text = generate_response(user_message, history)
     except Exception as e:
         tb = traceback.format_exc()
-        bot_text = f"⚠️ Error: {e}\n\n{tb}"
+        bot_text = f"⚠️ OhamLab encountered an error:\n\n{e}\n\n{tb}"
+
+    # Append OhamLab reply to chat history
+    chat_history.append({"role": "assistant", "content": f"**OhamLab:** {bot_text}"})
 
-    chat_history.append({"role": "assistant", "content": bot_text})
     return chat_history, ""
 
 
 def reset_chat():
+    """Resets the chat session."""
     return []
 
 
 # ---------------------------
-# 8. Gradio Chat UI
+# Gradio Chat UI
 # ---------------------------
 def build_ui():
     with gr.Blocks(
@@ -239,30 +248,32 @@ def build_ui():
     ) as demo:
 
        
-        # Chatbot (new format)
+
+        # Chatbot (branded to OhamLab)
         chatbot = gr.Chatbot(
+            label="💠 OhamLab Conversation",  # ✅ replaces "Chatbot" label
             height=520,
             elem_id="ohamlab",
-            type="messages",  # ✅ modern format
-            avatar_images=("🧑‍💻", "🤖"),
+            type="messages",  # ✅ OpenAI-style roles
+            avatar_images=("🧑‍💻", "💠"),  # user / OhamLab
         )
 
         # Input box (full width)
         with gr.Row():
             msg = gr.Textbox(
-                placeholder="Type your message here...",
+                placeholder="Ask OhamLab anything about R&D, AI, or engineering...",
                 lines=3,
                 scale=12,
                 show_label=False,
                 container=False,
             )
 
-        # Buttons (Send + Clear aligned below)
+        # Buttons (Send + Clear)
         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"])
 
-        # Wiring
+        # Wire actions
         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)
@@ -277,3 +288,4 @@ if __name__ == "__main__":
     print("🚀 Starting OhamLab Assistant...")
     demo = build_ui()
     demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+