Initial commit

app.py

@@ -25,7 +25,7 @@ logger.debug("Loading DeepSeek-Coder-6.7B-Instruct model")
 try:
     tokenizer = AutoTokenizer.from_pretrained("DeepSeek/DeepSeek-Coder-6.7B-Instruct", cache_dir="/tmp/hf_cache")
     model = AutoModelForCausalLM.from_pretrained("DeepSeek/DeepSeek-Coder-6.7B-Instruct", cache_dir="/tmp/hf_cache", torch_dtype=torch.float16, device_map="auto")
-    code_gen = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=1024, temperature=0.7, top_p=0.9)
+    code_gen = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=512, temperature=0.6, top_p=0.95)
     logger.debug("DeepSeek-Coder-6.7B-Instruct model loaded")
 except Exception as e:
     logger.error(f"Failed to load DeepSeek model: {e}")
@@ -101,33 +101,29 @@ def get_history():
 def debug_model(text, language):
     logger.debug(f"Debugging model with input: {text}, language: {language}")
     try:
-        prompt = f"""You are a coding assistant. The user provided this instruction in Persian: "{text}". Translate it to English, then write a complete, correct, and well-explained code in {language} based on the translated instruction. Return the English translation, the prompt, and the generated code."""
+        prompt = f"Write a complete, correct, and well-explained code in {language} to: {text}"
         logger.debug(f"Generated prompt: {prompt}")
-        result = code_gen(prompt, max_new_tokens=1024, temperature=0.7, top_p=0.9, do_sample=True)[0]['generated_text']
+        result = code_gen(prompt, max_new_tokens=512, temperature=0.6, top_p=0.95, do_sample=True)[0]['generated_text']
         logger.debug(f"Generated output: {result}")
-        return f"Debug: {result}"
+        return f"Debug: Prompt: {prompt}\nOutput: {result}"
     except Exception as e:
         logger.error(f"Debug error: {e}")
         return f"Debug error: {e}"
 
-# تولید پرامپت و کد
-def generate_code_and_prompt(text, language):
+# تولید کد
+def generate_code(text, language):
     logger.debug(f"Processing input: {text}, language: {language}")
     try:
-        # پرامپت برای DeepSeek که هم ترجمه کنه و هم کد تولید کنه
-        prompt = f"""You are a coding assistant. The user provided this instruction in Persian: "{text}". First, translate the instruction to English. Then, write a complete, correct, and well-explained code in {language} based on the translated instruction. Return the English translation followed by the code."""
+        prompt = f"Write a complete, correct, and well-explained code in {language} to: {text}"
         logger.debug(f"Generated prompt: {prompt}")
-        result = code_gen(prompt, max_new_tokens=1024, temperature=0.7, top_p=0.9, do_sample=True)[0]['generated_text']
+        result = code_gen(prompt, max_new_tokens=512, temperature=0.6, top_p=0.95, do_sample=True)[0]['generated_text']
         
-        # جدا کردن ترجمه و کد از خروجی
+        # استخراج کد از خروجی
         lines = result.split('\n')
-        translation = ""
         code = []
         is_code = False
         for line in lines:
-            if line.startswith("Translation:"):
-                translation = line.replace("Translation:", "").strip()
-            elif line.startswith("```"):
+            if line.startswith("```"):
                 is_code = not is_code
                 continue
             elif is_code:
@@ -138,26 +134,25 @@ def generate_code_and_prompt(text, language):
             logger.warning("Generated code is empty")
             code_output = "خطا: کد تولیدشده خالی است"
         
-        logger.debug(f"Translation: {translation}")
         logger.debug(f"Generated code: {code_output}")
         torch.cuda.empty_cache()
-        return translation, prompt, code_output
+        return prompt, code_output
     except Exception as e:
         logger.error(f"Code generation error: {e}")
-        return "", f"Error prompt: {text}", f"Code generation error: {e}"
+        return f"Error prompt: {text}", f"Code generation error: {e}"
 
 # تابع اصلی
 def christopher(user_input, language, show_history):
     logger.debug(f"Processing input: {user_input}, language: {language}, show_history: {show_history}")
     try:
         task_type = "نوشتن کد کامل"
-        # تولید ترجمه، پرامپت و کد
-        translation, prompt, response = generate_code_and_prompt(user_input, language)
+        # تولید پرامپت و کد
+        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, f"Translation: {translation}\nPrompt: {prompt}", response)
+        save_to_memory(user_input, task_type, prompt, response)
         history = get_history() if show_history else ""
         return response, history
     except Exception as e: