Update main.py

main.py

@@ -18,7 +18,7 @@ app = Flask(__name__)
 
 @app.route('/')
 def home():
-    return "ربات یکپارچه آلفا (نسخه پرو + دانلود فایل ۱۰۰٪ تضمینی + چت چندرسانه‌ای) روشن است! 🚀"
+    return "ربات یکپارچه آلفا (نسخه پرو + دانلود فایل ۱۰۰٪ تضمینی + چت چندرسانه‌ای + پرامپت‌نویس هوشمند) روشن است! 🚀"
 
 def run_flask():
     app.run(host="0.0.0.0", port=7860)
@@ -245,7 +245,7 @@ SPEAKERS = {
 user_states = {}
 
 
-# --- ۱. پردازش چت متنی و چندرسانه‌ای (ارتقا یافته) ---
+# --- ۱. پردازش چت متنی و چندرسانه‌ای ---
 async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=None):
     if not GEMINI_KEYS:
         await send_with_keyboard(client, chat_id, "❌ کلیدهای API جیمینای تنظیم نشده‌اند.", False)
@@ -255,7 +255,6 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
     
     history = user_states[chat_id].get("history",[])
 
-    # آماده‌سازی بخش‌های پیام کاربر (متن + فایل در صورت وجود)
     new_parts =[]
     
     if prompt:
@@ -283,14 +282,11 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
             }
         })
 
-    # افزودن پیام جدید به سابقه
     if history and history[-1]["role"] == "user":
-        # اگر پیام قبلی هم از کاربر بود، پیام جدید (فایل یا متن) را به آن اضافه کن
         history[-1]["parts"].extend(new_parts)
     else:
         history.append({"role": "user", "parts": new_parts})
 
-    # تغییر تاریخچه به ۴۰ مورد برای ذخیره کردن ۲۰ سوال و جواب (۲۰ پیام رفت و برگشت)
     if len(history) > 40:
         history = history[-40:]
         if history[0]["role"] == "model":
@@ -361,14 +357,57 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
         await send_with_keyboard(client, chat_id, "❌ خطایی در ارسال پیام رخ داد.", False)
 
 
-# --- ۲. پردازش ساخت عکس ---
+# --- ۲. پردازش ساخت عکس (ارتقا یافته با پرامپت‌نویس هوشمند) ---
 async def process_image(client, chat_id, prompt):
     if not HF_TOKENS:
         await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
         return
 
-    proc_msg = await send_with_keyboard(client, chat_id, "🎨 در حال طراحی عکس...\n(ممکن است چند ثانیه زمان ببرد)", False)
+    proc_msg = await send_with_keyboard(client, chat_id, "✨ در حال ترجمه و بهینه‌سازی پرامپت شما توسط جیمینای...\n(تبدیل به پرامپت حرفه‌ای)", False)
     
+    # مرحله اول: ارسال پرامپت به جیمینای برای ترجمه و بهینه‌سازی
+    enhanced_prompt = prompt  # مقدار پیش‌فرض اگر جیمینای خطا داد
+    if GEMINI_KEYS:
+        keys_to_try_gemini = GEMINI_KEYS.copy()
+        random.shuffle(keys_to_try_gemini)
+        
+        # پرامپت سیستم برای جیمینای
+        gemini_sys_prompt = (
+            "You are an expert AI image generation prompt engineer. "
+            "Translate the following user input to English, and enhance it with high-quality, "
+            "highly detailed, 4k resolution, cinematic lighting, and visually striking descriptive keywords "
+            "suitable for Midjourney or Stable Diffusion. "
+            "Return ONLY the final English prompt string, with absolutely no extra text or explanations. "
+            f"User input: {prompt}"
+        )
+        
+        async with aiohttp.ClientSession() as session:
+            for key in keys_to_try_gemini:
+                url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={key}"
+                payload = {"contents": [{"parts":[{"text": gemini_sys_prompt}]}], "generationConfig": {"temperature": 0.7}}
+                try:
+                    async with session.post(url, json=payload, timeout=20) as response:
+                        if response.status == 200:
+                            data = await response.json()
+                            enhanced_prompt = data["candidates"][0]["content"]["parts"][0]["text"].strip()
+                            break
+                except Exception:
+                    continue
+
+    # پاک کردن پیام قبلی و اطلاع به کاربر درباره شروع طراحی
+    try:
+        if proc_msg:
+            msg_id = getattr(proc_msg, 'message_id', None)
+            if isinstance(proc_msg, dict):
+                msg_id = proc_msg.get('message_update', {}).get('message_id') or proc_msg.get('message_id')
+            if msg_id:
+                await client.delete_messages(chat_id,[msg_id])
+    except Exception: pass
+    
+    short_preview = enhanced_prompt[:150] + "..." if len(enhanced_prompt) > 150 else enhanced_prompt
+    proc_msg = await send_with_keyboard(client, chat_id, f"🎨 در حال طراحی عکس...\n\n📝 پرامپت حرفه‌ای ساخته شده:\n`{short_preview}`\n\n(ممکن است چند ثانیه زمان ببرد)", False)
+
+    # مرحله دوم: ساخت عکس با پرامپت بهینه‌شده
     keys_to_try = HF_TOKENS.copy()
     random.shuffle(keys_to_try)
     generated_image = None
@@ -377,7 +416,8 @@ async def process_image(client, chat_id, prompt):
     for token in keys_to_try:
         try:
             hf_client = AsyncInferenceClient(provider="fal-ai", api_key=token)
-            generated_image = await hf_client.text_to_image(prompt, model="Tongyi-MAI/Z-Image-Turbo")
+            # اینجا پرامپت بهینه‌شده ارسال می‌شود
+            generated_image = await hf_client.text_to_image(enhanced_prompt, model="Tongyi-MAI/Z-Image-Turbo")
             break 
         except Exception as e:
             last_error_log = str(e)
@@ -402,7 +442,7 @@ async def process_image(client, chat_id, prompt):
         rgb_im.save(file_name, format="JPEG", quality=100)
         
         await asyncio.sleep(1)
-        caption_text = "🎨 تصویر شما آماده شد!\n(مدل: Z-Image-Turbo)"
+        caption_text = f"🎨 تصویر شما با پرامپت هوشمند آماده شد!\n\n✨ ایده اولیه شما: {prompt}\n(مدل: Z-Image-Turbo)"
         
         upload_result = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
         
@@ -637,7 +677,7 @@ else:
 
             if user_text_str in["/image", "🎨 عکس", "ساخت تصاویر با هوش مصنوعی🎨"]:
                 user_states[chat_id] = {"mode": "image_waiting_for_text", "text": "", "history":[]}
-                await send_with_keyboard(client, chat_id, "🎨 شما وارد بخش **ساخت عکس** شدید.\n\nلطفاً متنی که می‌خواهید به عکس تبدیل شود را توصیف کنید:\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                await send_with_keyboard(client, chat_id, "🎨 شما وارد بخش **ساخت عکس پیشرفته** شدید.\n\n💡 نکته: کافیست تصویر مورد نظرتان را به سادگی توصیف کنید، من خودم آن را به یک پرامپت انگلیسی و حرفه‌ای تبدیل می‌کنم و می‌سازم!\n\nمتن خود را ارسال کنید:\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/tts", "🎙️ صدا", "تبدیل متن به صدا با هوش مصنوعی🎙️"]:
@@ -662,13 +702,12 @@ else:
                 elif user_text_str: await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا از کیبورد پایین، بخش مورد نظرتان را انتخاب کنید:", True)
                 return
 
-            # --- مسیریابی و هندل کردن حالت چت (ارتقا یافته برای پشتیبانی از ارسال فایل) ---
+            # --- مسیریابی ---
             elif current_mode == "chat":
                 if is_file:
                     await send_with_keyboard(client, chat_id, "📥 در حال دانلود فایل برای چت هوشمند...", False)
                     try:
                         file_bytes = await helper_download_file(client, msg_obj)
-                        # اگر کاربر همراه عکس کپشن نوشته باشد ارسال می‌شود، در غیر اینصورت ربات متوجه می‌شود فایل فرستاده
                         asyncio.create_task(process_gemini(client, chat_id, user_text_str, file_bytes=file_bytes, file_name=file_name))
                     except Exception as dl_err:
                         err_trace = traceback.format_exc()
@@ -731,7 +770,6 @@ else:
                     await send_with_keyboard(client, chat_id, f"❌ خطا در دانلود فایل!\n\n🔴 لاگ برنامه‌نویس:\n`{err_trace[-300:]}`\n{str(dl_err)}", False)
                 return
 
-            # --- بخش تحلیل فایل اختصاصی ---
             elif current_mode == "file_waiting_for_file":
                 if not is_file:
                     await send_with_keyboard(client, chat_id, "⚠️ لطفاً ابتدا یک فایل (تصویر، ویدیو، PDF و...) ارسال کنید.", False)