Update app.py

app.py

@@ -58,16 +58,20 @@ def gemini_response(prompt):
     except Exception as e:
         return f"⚠️ Gemini error: {e}"
 
-def chat(prompt, history=[], theme="Dark"):
-    history.append((f"🧑 {prompt}", None))
-    time.sleep(0.5)
-    reply_raw = gemini_response(prompt)
-    reply_md = markdown(f"🤖 {reply_raw}")
-    history[-1] = (history[-1][0], reply_md)
+# ======================
+# 🌙 CHAT FUNCTION
+# ======================
+
+def chat_fn(prompt, history):
+    history = history or []
+    history.append({"role": "user", "content": prompt})
+    time.sleep(0.5)  # simulate typing
+    reply = gemini_response(prompt)
+    history.append({"role": "bot", "content": reply})
     return history, history
 
 # ======================
-# 🌙 UI CSS
+# 🌟 UI CSS
 # ======================
 
 css = """
@@ -86,6 +90,9 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
 
     gr.Markdown("## 🔐 Welcome to ZeppFusion Login System")
 
+    # ======================
+    # Tabs: Login / Signup
+    # ======================
     with gr.Tabs():
 
         # -------------------
@@ -106,51 +113,46 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
             reg_status = gr.Textbox(label="Status", interactive=False)
             reg_btn = gr.Button("Signup")
 
-        # Hidden login flag
-        logged_in = gr.State(False)
-
-        # -------------------
-        # CHAT UI (after login)
-        # -------------------
-        with gr.Group(visible=False) as chat_ui:
-            with gr.Row():
-                gr.Image(value=logo_path, show_label=False)
-                gr.Markdown("## ZeppFusion AI")
-
-            chatbot = gr.Chatbot()
-            msg = gr.Textbox(placeholder="Таны асуулт...", lines=1)
-            clear = gr.Button("🗑️ Clear History")
+    # ======================
+    # Chat UI (hidden initially)
+    # ======================
+    with gr.Group(visible=False) as chat_ui:
+        with gr.Row():
+            gr.Image(value=logo_path, show_label=False)
+            gr.Markdown("## ZeppFusion AI Chat")
+        chatbot = gr.Chatbot(type="messages")
+        msg = gr.Textbox(placeholder="Таны асуулт...", lines=1)
+        clear = gr.Button("🗑️ Clear History")
+        msg.submit(chat_fn, inputs=[msg, chatbot], outputs=[chatbot, chatbot])
+        clear.click(lambda: [], None, chatbot)
+
+    # ======================
+    # BUTTON ACTIONS
+    # ======================
+
+    # Signup
+    reg_btn.click(
+        signup,
+        inputs=[reg_user, reg_pw],
+        outputs=reg_status
+    )
+
+    # Login
+    def login_action(username, password):
+        res = login(username, password)
+        if res == "SUCCESS":
+            return "✅ Амжилттай нэвтэрлээ!", gr.update(visible=True)
+        return res, gr.update(visible=False)
+
+    log_btn.click(
+        login_action,
+        inputs=[log_user, log_pw],
+        outputs=[log_status, chat_ui]
+    )
 
-            msg.submit(chat, [msg, chatbot, None], [chatbot, chatbot])
-            clear.click(lambda: [], None, chatbot)
-
-        # -------------------
-        # BUTTON ACTIONS
-        # -------------------
-
-        # Signup action
-        reg_btn.click(
-            signup,
-            inputs=[reg_user, reg_pw],
-            outputs=reg_status
-        )
-
-        # Login action
-        def login_action(username, password):
-            res = login(username, password)
-            if res == "SUCCESS":
-                return "✅ Амжилттай нэвтэрлээ!", True, gr.update(visible=True)
-            return res, False, gr.update(visible=False)
-
-        log_btn.click(
-            login_action,
-            inputs=[log_user, log_pw],
-            outputs=[log_status, logged_in, chat_ui]
-        )
-
-# -----------------------
+# ======================
 # RUN
-# -----------------------
+# ======================
 
 if __name__ == "__main__":
     app.launch()