Update app.py

app.py

@@ -158,65 +158,91 @@ def generate_response(user_input, history, mode="chat"):
 # ---------------------------
 # 7. Gradio Chat UI
 # ---------------------------
-import traceback
-import gradio as gr
-
-# ---------------------------
-# Chat Logic
-# ---------------------------
 import gradio as gr
+from typing import List, Tuple
 
-def chat_with_model(message, history):
+# --- Core Chat Logic ---
+def chat_with_model(message: str, history: List[Tuple[str, str]]):
     if history is None:
         history = []
-    # Simulate bot response
-    response = f"OhamLab says: {message[::-1]}"  # replace with actual model call
-    history.append(("user", message))
-    history.append(("bot", response))
+
+    system_message = "You are OhamLab AI, a professional assistant for research and engineering."
+
+    # Build structured message history
+    messages = [{"role": "system", "content": system_message}]
+    for user_msg, bot_msg in history:
+        if user_msg:
+            messages.append({"role": "user", "content": user_msg})
+        if bot_msg:
+            messages.append({"role": "assistant", "content": bot_msg})
+
+    # Add current user input
+    messages.append({"role": "user", "content": message})
+
+    # Placeholder: Replace with your model inference (RAG, etc.)
+    bot_reply = f"OhamLab: {message[::-1]}"  # simulate intelligent response
+
+    # Append to history (user msg, bot reply)
+    history.append((message, bot_reply))
     return history, ""
 
+
 def reset_chat():
     return []
 
+
+# --- Professional OhamLab Chat UI ---
 def build_ui():
     with gr.Blocks(
         theme=gr.themes.Soft(),
         css="""
+        /* Hide share/delete icons */
+        #ohamlab .wrap.svelte-1lcyrj3 > div > div > button {
+            display: none !important;
+        }
+
+        /* Chat alignment and bubble styling */
         #ohamlab .message.user {
-            background-color: #e6f0ff !important;
-            border-radius: 14px !important;
-            color: #003366 !important;
+            background-color: #0047AB !important;
+            color: white !important;
+            border-radius: 16px !important;
+            align-self: flex-end !important;
+            text-align: right !important;
+            margin-left: 20%;
         }
+
         #ohamlab .message.bot {
-            background-color: #f8f9fa !important;
-            border-radius: 14px !important;
+            background-color: #f2f2f2 !important;
             color: #111 !important;
+            border-radius: 16px !important;
+            align-self: flex-start !important;
+            text-align: left !important;
+            margin-right: 20%;
         }
-        #ohamlab .chatbot .wrap.svelte-1lcyrj3 > div > div > button {
-            display: none !important; /* hide share/delete icons */
-        }
-        #ohamlab .bot-icon {
-            background: linear-gradient(90deg, #0047AB, #007FFF);
-            color: white;
-            border-radius: 50%;
-            padding: 6px 10px;
-            font-weight: 600;
-            font-size: 14px;
-        }
+
+        /* General layout tweaks */
         .message-box {
             width: 100%;
         }
         """,
     ) as demo:
-        
-       
+        gr.Markdown(
+            """
+            <div style='text-align:center;'>
+                <h2 style='color:#0047AB;'>🤖 OhamLab AI Assistant</h2>
+                <p style='color:gray;'>Your trusted partner in advanced R&D and AI innovation.</p>
+            </div>
+            """
+        )
 
         chatbot = gr.Chatbot(
+            label="💠 OhamLab Conversation",
+            elem_id="ohamlab",
             label=None,
             height=540,
-            elem_id="ohamlab",
             bubble_full_width=False,
-            avatar_images=[None, None],  # disable circular icons
+            show_copy_button=False,
+            avatar_images=[None, None],
         )
 
         with gr.Row(equal_height=False):
@@ -231,14 +257,15 @@ def build_ui():
             send = gr.Button("Send", variant="primary")
             clear = gr.Button("Clear")
 
-        # event bindings
         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
 
+
 if __name__ == "__main__":
     demo = build_ui()
     demo.launch(server_name="0.0.0.0", server_port=7860)
 
+