Initial commit

app.py

@@ -20,15 +20,15 @@ if not hf_token:
 login(token=hf_token)
 logger.debug("Hugging Face login successful")
 
-# لود مدل Gemma 3 4B
-logger.debug("Loading google/gemma-3-4b-it model")
+# لود مدل Gemma 3 1B
+logger.debug("Loading google/gemma-3-1b-it model")
 try:
-    tokenizer = AutoTokenizer.from_pretrained("google/gemma-3-4b-it", cache_dir="/tmp/hf_cache")
-    model = AutoModelForCausalLM.from_pretrained("google/gemma-3-4b-it", cache_dir="/tmp/hf_cache", torch_dtype=torch.bfloat16, device_map="auto")
+    tokenizer = AutoTokenizer.from_pretrained("google/gemma-3-1b-it", cache_dir="/tmp/hf_cache")
+    model = AutoModelForCausalLM.from_pretrained("google/gemma-3-1b-it", cache_dir="/tmp/hf_cache", torch_dtype=torch.bfloat16, device_map="auto")
     code_gen = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=512, temperature=0.6, top_p=0.95)
-    logger.debug("google/gemma-3-4b-it model loaded")
+    logger.debug("google/gemma-3-1b-it model loaded")
 except Exception as e:
-    logger.error(f"Failed to load Gemma 3 model: {e}")
+    logger.error(f"Failed to load Gemma 3 1B model: {e}")
     raise
 
 # آزادسازی حافظه
@@ -71,7 +71,7 @@ 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}")
-        pass  # نذار خطای دیتابیس جلوی ادامه کار رو بگیره
+        pass
 
 # دریافت تاریخچه
 def get_history():
@@ -146,12 +146,10 @@ def christopher(user_input, language, show_history):
     logger.debug(f"Processing input: {user_input}, language: {language}, show_history: {show_history}")
     try:
         task_type = "نوشتن کد کامل"
-        # تولید پرامپت و کد
         prompt, response = generate_code(user_input, language)
         if "error" in response.lower():
             history = get_history() if show_history else ""
             return response, history
-        # ذخیره تو دیتابیس (اختیاری)
         save_to_memory(user_input, task_type, prompt, response)
         history = get_history() if show_history else ""
         return response, history