Update app.py

app.py

@@ -164,128 +164,84 @@ import gradio as gr
 # ---------------------------
 # 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, ""
-
-    # 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"⚠️ 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}"})
-
-    return chat_history, ""
+import gradio as gr
 
+def chat_with_model(message, history):
+    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))
+    return history, ""
 
 def reset_chat():
-    """Resets the chat session."""
     return []
 
-
-# ---------------------------
-# Gradio Chat UI
-# ---------------------------
 def build_ui():
     with gr.Blocks(
-        theme=gr.themes.Soft(primary_hue="indigo"),
+        theme=gr.themes.Soft(),
         css="""
-            #ohamlab {
-                height: 550px !important;
-                border-radius: 14px;
-                box-shadow: 0 2px 12px rgba(0,0,0,0.1);
-                background: white;
-            }
-            .gradio-container {
-                max-width: 900px !important;
-                margin: auto;
-                padding-top: 1.5rem;
-            }
-            textarea {
-                resize: none !important;
-                border-radius: 12px !important;
-                border: 1px solid #d1d5db !important;
-                box-shadow: 0 1px 3px rgba(0,0,0,0.08);
-            }
-            button.primary {
-                background-color: #4f46e5 !important;
-                color: white !important;
-                border-radius: 10px !important;
-                padding: 0.6rem 1.4rem !important;
-                font-weight: 600;
-                transition: all 0.2s ease-in-out;
-            }
-            button.primary:hover {
-                background-color: #4338ca !important;
-            }
-            button.secondary {
-                background-color: #f3f4f6 !important;
-                border-radius: 10px !important;
-                color: #374151 !important;
-                font-weight: 500;
-                transition: all 0.2s ease-in-out;
-            }
-            button.secondary:hover {
-                background-color: #e5e7eb !important;
-            }
-        """
+        #ohamlab .message.user {
+            background-color: #e6f0ff !important;
+            border-radius: 14px !important;
+            color: #003366 !important;
+        }
+        #ohamlab .message.bot {
+            background-color: #f8f9fa !important;
+            border-radius: 14px !important;
+            color: #111 !important;
+        }
+        #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;
+        }
+        .message-box {
+            width: 100%;
+        }
+        """,
     ) as demo:
+        
+        gr.Markdown(
+            "<h2 style='text-align:center; color:#0047AB;'>💬 OhamLab AI Assistant</h2>"
+            "<p style='text-align:center; color:gray;'>Your professional research and engineering companion.</p>"
+        )
 
-       
-
-        # Chatbot (branded to OhamLab)
         chatbot = gr.Chatbot(
-            label="💠 OhamLab Conversation",  # ✅ replaces "Chatbot" label
-            height=520,
+            label=None,
+            height=540,
             elem_id="ohamlab",
-            type="messages",  # ✅ OpenAI-style roles
-            avatar_images=("🧑‍💻", "💠"),  # user / OhamLab
+            bubble_full_width=False,
+            avatar_images=[None, None],  # disable circular icons
         )
 
-        # Input box (full width)
-        with gr.Row():
+        with gr.Row(equal_height=False):
             msg = gr.Textbox(
-                placeholder="Ask OhamLab anything about R&D, AI, or engineering...",
+                placeholder="Type your message here...",
                 lines=3,
-                scale=12,
                 show_label=False,
-                container=False,
+                elem_classes="message-box",
             )
 
-        # 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"])
+        with gr.Row():
+            send = gr.Button("Send", variant="primary")
+            clear = gr.Button("Clear")
 
-        # Wire actions
+        # 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
 
-
-# ---------------------------
-# Entrypoint
-# ---------------------------
 if __name__ == "__main__":
-    print("🚀 Starting OhamLab Assistant...")
     demo = build_ui()
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=False)
+    demo.launch(server_name="0.0.0.0", server_port=7860)