Update app.py

app.py

@@ -1,7 +1,6 @@
-# app.py - FINAL VERSION WITH LANGUAGE DETECTION
+# app.py - FINAL VERSION WITH TiDB INTEGRATION
 import os
 import uvicorn
-import json
 from fastapi import FastAPI, Form
 from fastapi.responses import HTMLResponse, JSONResponse
 from groq import Groq
@@ -10,26 +9,8 @@ from groq import Groq
 app = FastAPI()
 client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
-# Configuration & Persistent Memory Setup
+# Configuration
 USERNAME = "Sanjay"
-MEMORY_FILE = "memory.json"
-
-def load_memory():
-    if os.path.exists(MEMORY_FILE):
-        try:
-            with open(MEMORY_FILE, 'r') as f:
-                data = json.load(f)
-                return data if "history" in data else {"history": []}
-        except Exception as e:
-            return {"history": []}
-    return {"history": []}
-
-def save_memory(chat_history):
-    try:
-        with open(MEMORY_FILE, 'w') as f:
-            json.dump({"history": chat_history}, f, indent=4)
-    except Exception as e:
-        print(f"Memory Save Error: {e}")
 
 # UI REMAINS EXACTLY SAME
 HTML_UI = '''
@@ -71,8 +52,8 @@ HTML_UI = '''
             <div class="nav-item reset-btn" onclick="confirmReset()"><i class="fas fa-trash-alt"></i> Reset Memory</div>
             <div class="nav-item"><i class="fas fa-cog"></i> Settings</div>
         </div>
-    </div>
-    <div class="main-chat">
+</div>
+<div class="main-chat">
         <div id="chat-log" class="chat-box"></div>
         <div class="input-area">
             <div class="input-container">
@@ -121,23 +102,33 @@ async def get_ui():
 
 @app.post("/reset")
 async def reset():
-    save_memory([])
+    # TiDB reset logic (baad mein add karna)
     return {"message": "Memory clear ho gayi!"}
 
 @app.post("/chat")
 async def chat(message: str = Form(...)):
     # ✅ LANGUAGE DETECTION
     from language_detector import detect_input_language, get_system_prompt
+    from memory_db import tidb_memory  # ✅ TiDB IMPORT
+    
     lang_mode = detect_input_language(message)
     system_prompt = get_system_prompt(lang_mode, USERNAME)
     
-    memory_data = load_memory()
-    history = memory_data.get("history", [])[-10:]
+    # ✅ GET HISTORY FROM TiDB (JSON ki jagah)
+    recent_chats = []
+    try:
+        recent_chats = tidb_memory.get_recent_chats(limit=10)
+    except Exception as e:
+        print(f"TiDB history fetch error: {e}")
     
+    # Build messages
     api_messages = [{"role": "system", "content": system_prompt}]
-    for chat_pair in history:
-        api_messages.append({"role": "user", "content": chat_pair["u"]})
-        api_messages.append({"role": "assistant", "content": chat_pair["a"]})
+    
+    # Add history from TiDB
+    for chat_row in recent_chats:
+        user_input, ai_response, _ = chat_row
+        api_messages.append({"role": "user", "content": user_input})
+        api_messages.append({"role": "assistant", "content": ai_response})
     
     api_messages.append({"role": "user", "content": message})
     
@@ -149,8 +140,13 @@ async def chat(message: str = Form(...)):
             max_tokens=800
         )
         ai_response = completion.choices[0].message.content.strip()
-        history.append({"u": message, "a": ai_response})
-        save_memory(history)
+        
+        # ✅ SAVE TO TiDB
+        try:
+            tidb_memory.save_chat(message, ai_response, lang_mode)
+        except Exception as e:
+            print(f"TiDB save error: {e}")
+        
         return {"response": ai_response}
     except Exception as e:
         return {"response": f"Error: {str(e)}"}