Initial commit

app.py

@@ -47,11 +47,14 @@ import gc
 gc.collect()
 torch.cuda.empty_cache()
 
-# ایجاد دیتابیس در حافظه
+# مسیر فایل دیتابیس
+DB_PATH = "chris.db"
+
+# ایجاد دیتابیس
 def init_db():
-    logger.debug("Initializing in-memory database")
+    logger.debug("Initializing database")
     try:
-        with sqlite3.connect(":memory:") as conn:
+        with sqlite3.connect(DB_PATH) as conn:
             cursor = conn.cursor()
             cursor.execute('''CREATE TABLE IF NOT EXISTS CHRIS (
                                 id INTEGER PRIMARY KEY AUTOINCREMENT,
@@ -71,7 +74,7 @@ def init_db():
 def save_to_memory(user_input, task_type, prompt, response):
     logger.debug("Saving to database")
     try:
-        with sqlite3.connect(":memory:") as conn:
+        with sqlite3.connect(DB_PATH) as conn:
             cursor = conn.cursor()
             cursor.execute("INSERT INTO CHRIS (user_input, task_type, prompt, response, timestamp) VALUES (?, ?, ?, ?, ?)",
                            (user_input, task_type, prompt, response, datetime.now().isoformat()))
@@ -79,13 +82,14 @@ def save_to_memory(user_input, task_type, prompt, response):
             logger.debug("Saved to DB")
     except Exception as e:
         logger.error(f"DB Save Error: {e}")
-        raise
+        # نذار خطای دیتابیس جلوی ادامه کار رو بگیره
+        pass
 
 # دریافت تاریخچه
 def get_history():
     logger.debug("Retrieving history from database")
     try:
-        with sqlite3.connect(":memory:") as conn:
+        with sqlite3.connect(DB_PATH) as conn:
             cursor = conn.cursor()
             cursor.execute("SELECT id, user_input, task_type, prompt, response, timestamp FROM CHRIS ORDER BY timestamp DESC")
             rows = cursor.fetchall()
@@ -109,12 +113,10 @@ def get_history():
 def debug_models(text, language):
     logger.debug(f"Debugging models with input: {text}, language: {language}")
     try:
-        # تست مدل ترجمه
         translated = translate_gen(text, max_length=300)[0]['translation_text']
         logger.debug(f"Translated text: {translated}")
         prompt = f"Write a complete, correct, and well-explained code in {language} to: {translated}"
         logger.debug(f"Generated prompt: {prompt}")
-        # تست مدل کدنویسی
         code_output = code_gen(prompt, max_length=512, num_beams=5)[0]['generated_text']
         logger.debug(f"Generated code: {code_output}")
         return f"Debug: Translation: {translated}\nPrompt: {prompt}\nCode: {code_output}"
@@ -159,6 +161,7 @@ def christopher(user_input, language, show_history):
             return prompt, history
         # تولید کد
         response = generate_code(prompt)
+        # ذخیره تو دیتابیس (اختیاری)
         save_to_memory(user_input, task_type, prompt, response)
         history = get_history() if show_history else ""
         return response, history