Update app.py

app.py

@@ -1,33 +1,42 @@
 import os
 import requests
 import gradio as gr
+import sqlite3
 from openai import OpenAI
 
-# 🔑 API key-г environment variable-аас уншина
+# 🔑 API keys
 openai_api_key = os.getenv("OPENAI_API_KEY")
 gemini_api_key = os.getenv("GEMINI_API_KEY")
 
-# Баталгаажуулах
 print("OPENAI_API_KEY =", openai_api_key)
 print("GEMINI_API_KEY =", gemini_api_key)
 
-# OpenAI client
 if openai_api_key is None:
     raise ValueError("⚠️ OPENAI_API_KEY тохируулна уу!")
+
 client = OpenAI(api_key=openai_api_key)
 
-# ChatGPT (gpt-5) хариу
+# SQLite db setup
+conn = sqlite3.connect("chat_history.db", check_same_thread=False)
+c = conn.cursor()
+c.execute("""
+CREATE TABLE IF NOT EXISTS history (
+    id INTEGER PRIMARY KEY AUTOINCREMENT,
+    user_input TEXT,
+    fusion_output TEXT
+)
+""")
+conn.commit()
+
+# ChatGPT (gpt-5)
 def chatgpt_response(prompt):
     try:
-        response = client.responses.create(
-            model="gpt-5",
-            input=prompt
-        )
+        response = client.responses.create(model="gpt-5", input=prompt)
         return response.output_text
     except Exception as e:
         return f"⚠️ ChatGPT error: {e}"
 
-# Gemini хариу
+# Gemini
 def gemini_response(prompt):
     if gemini_api_key is None:
         return "⚠️ GEMINI_API_KEY тохируулаагүй байна!"
@@ -41,26 +50,25 @@ def gemini_response(prompt):
     except Exception as e:
         return f"⚠️ Gemini error: {e}"
 
-# Fusion logic (ChatGPT quota дууссан үед Gemini fallback)
-def fusion_ai(prompt):
+# Fusion + summary
+def fusion_ai(prompt, chat_history=[]):
     gpt_reply = chatgpt_response(prompt)
     gemini_reply = gemini_response(prompt)
 
     if "ChatGPT error" in gpt_reply.lower():
-        return f"⚠️ ChatGPT quota дууссан тул Gemini хариу:\n\n{gemini_reply}"
+        combined = f"⚠️ ChatGPT quota дууссан тул Gemini хариу:\n\n{gemini_reply}"
+    else:
+        # Fusion summary: урт хариултаас сонгох
+        combined = f"ChatGPT 🧠: {gpt_reply}\nGemini 🌐: {gemini_reply}"
 
-    return f"""
-<div style="background-color:#0d1117; padding:20px; border-radius:10px; color:#e6edf3; font-family:Segoe UI">
-    <h3>🧩 <span style="color:#00d9ff">Fusion Summary</span></h3>
-    <p><b>ChatGPT 🧠:</b> {gpt_reply}</p>
-    <p><b>Gemini 🌐:</b> {gemini_reply}</p>
-    <hr style="border:1px solid #222;">
-    <h4>✨ <span style="color:#9b59b6">ZeppFusion Combined Insight:</span></h4>
-    <p>{gpt_reply if len(gpt_reply) > len(gemini_reply) else gemini_reply}</p>
-</div>
-"""
+    # Update SQLite history
+    c.execute("INSERT INTO history (user_input, fusion_output) VALUES (?, ?)", (prompt, combined))
+    conn.commit()
 
-# Gradio Interface
+    chat_history.append((prompt, combined))
+    return chat_history, chat_history
+
+# Gradio interface
 css = """
 .gradio-container {background-color: #0d1117 !important; color: #e6edf3;}
 h1, h3, h4, label {color: #00d9ff !important;}
@@ -68,15 +76,21 @@ textarea, input {background-color:#161b22 !important; color:#fff !important; bor
 button {background-color:#00d9ff !important; color:#000 !important; font-weight:bold; border-radius:10px;}
 """
 
-app = gr.Interface(
-    fn=fusion_ai,
-    inputs=gr.Textbox(label="🗨️ Та асуултаа бичнэ үү", placeholder="Жишээ нь: AI гэж юу вэ?"),
-    outputs=gr.HTML(label="💡 ZeppFusion AI-ийн хариулт"),
-    title="🚀 ZeppFusion AI",
-    description="ChatGPT (gpt-5) 🧠 + Gemini 🌐 хослуулсан Fusion AI chatbot — Where Intelligence Unites.",
-    theme="soft",
-    css=css,
-)
+with gr.Blocks(css=css, theme=gr.themes.Soft()) as app:
+    gr.Markdown("## 🚀 ZeppFusion AI — Persistent Fusion Chat")
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(placeholder="Таны асуулт...")
+    clear = gr.Button("🗑️ Clear History")
+
+    # Submit: add to history
+    msg.submit(fusion_ai, [msg, chatbot], [chatbot, chatbot])
+    
+    # Clear button: delete SQLite history
+    def clear_history():
+        c.execute("DELETE FROM history")
+        conn.commit()
+        return []
+    clear.click(clear_history, None, chatbot)
 
 if __name__ == "__main__":
     app.launch()