Update app.py

app.py

@@ -62,13 +62,15 @@ def gemini_response(prompt):
 # 🌙 CHAT FUNCTION
 # ======================
 
-def chat_fn(prompt, history):
-    history = history or []
-    history.append({"role": "user", "content": prompt})
+def chat_fn(message, chat_history):
+    chat_history = chat_history or []
+    # Хэрэглэгчийн мессеж нэмэх
+    chat_history.append({"role": "user", "content": message})
     time.sleep(0.5)  # simulate typing
-    reply = gemini_response(prompt)
-    history.append({"role": "bot", "content": reply})
-    return history, history
+    # Gemini reply нэмэх
+    reply = gemini_response(message)
+    chat_history.append({"role": "assistant", "content": reply})
+    return chat_history, chat_history
 
 # ======================
 # 🌟 UI CSS
@@ -90,32 +92,22 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
 
     gr.Markdown("## 🔐 Welcome to ZeppFusion Login System")
 
-    # ======================
     # Tabs: Login / Signup
-    # ======================
     with gr.Tabs() as tabs:
 
-        # -------------------
-        # LOGIN TAB
-        # -------------------
         with gr.Tab("Login") as login_tab:
             log_user = gr.Textbox(label="Username")
             log_pw = gr.Textbox(label="Password", type="password")
             log_status = gr.Textbox(label="Status", interactive=False)
             log_btn = gr.Button("Login")
 
-        # -------------------
-        # SIGNUP TAB
-        # -------------------
         with gr.Tab("Signup"):
             reg_user = gr.Textbox(label="New Username")
             reg_pw = gr.Textbox(label="New Password", type="password")
             reg_status = gr.Textbox(label="Status", interactive=False)
             reg_btn = gr.Button("Signup")
 
-    # ======================
     # Chat UI (hidden initially)
-    # ======================
     with gr.Group(visible=False) as chat_ui:
         with gr.Row():
             gr.Image(value=logo_path, show_label=False)
@@ -141,7 +133,7 @@ with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
     def login_action(username, password):
         res = login(username, password)
         if res == "SUCCESS":
-            # Hide login/signup tabs & show chat UI
+            # Login/signup tab-г hide хийж, chat UI-г гаргах
             return "✅ Амжилттай нэвтэрлээ!", gr.update(visible=False), gr.update(visible=True)
         return res, gr.update(visible=True), gr.update(visible=False)