Update main.py

main.py

@@ -12,6 +12,7 @@ import json
 import datetime
 import string
 import uuid
+import concurrent.futures
 from flask import Flask
 from rubpy.bot import BotClient, filters
 
@@ -212,7 +213,28 @@ def home():
     return "ربات یکپارچه آلفا (نسخه پرو + مدیریت اشتراک نامحدود + دعوت دوستان + سیستم ضد باگ) روشن است! 🚀"
 
 def run_flask():
-    app.run(host="0.0.0.0", port=7860)
+    app.run(host="0.0.0.0", port=7860, threaded=True)
+
+
+# --- توابع کمکی برای جلوگیری از بلاک شدن Event Loop ---
+def sync_save_image(image, file_name):
+    rgb_im = image.convert('RGB')
+    rgb_im.save(file_name, format="JPEG", quality=100)
+
+def sync_write_file(file_name, data):
+    with open(file_name, "wb") as f:
+        f.write(data)
+
+def sync_read_file(file_name):
+    with open(file_name, 'rb') as f:
+        return f.read()
+
+def sync_combine_audio(current_audio, new_bytes):
+    audio_segment = AudioSegment.from_file(io.BytesIO(new_bytes))
+    return current_audio + audio_segment
+
+def sync_export_audio(audio_obj, file_name):
+    audio_obj.export(file_name, format="mp3")
 
 
 # --- ساختار کیبورد آپدیت شده ---
@@ -368,7 +390,7 @@ async def helper_download_file(client, msg_obj):
         try:
             await client.download_file(msg_obj, temp_name)
             if os.path.exists(temp_name):
-                with open(temp_name, 'rb') as f: data = f.read()
+                data = await asyncio.to_thread(sync_read_file, temp_name)
                 os.remove(temp_name)
                 return data
         except Exception as e: errors.append(f"تلاش ۱: {e}")
@@ -377,7 +399,7 @@ async def helper_download_file(client, msg_obj):
             try:
                 await client.download_file(file_obj, temp_name)
                 if os.path.exists(temp_name):
-                    with open(temp_name, 'rb') as f: data = f.read()
+                    data = await asyncio.to_thread(sync_read_file, temp_name)
                     os.remove(temp_name)
                     return data
             except Exception as e: errors.append(f"تلاش ۲: {e}")
@@ -386,7 +408,7 @@ async def helper_download_file(client, msg_obj):
             try:
                 await client.download_file(file_id, file_name=temp_name)
                 if os.path.exists(temp_name):
-                    with open(temp_name, 'rb') as f: data = f.read()
+                    data = await asyncio.to_thread(sync_read_file, temp_name)
                     os.remove(temp_name)
                     return data
             except Exception as e: errors.append(f"تلاش ۳: {e}")
@@ -394,7 +416,7 @@ async def helper_download_file(client, msg_obj):
             try:
                 await client.download_file(file_id, save_as=temp_name)
                 if os.path.exists(temp_name):
-                    with open(temp_name, 'rb') as f: data = f.read()
+                    data = await asyncio.to_thread(sync_read_file, temp_name)
                     os.remove(temp_name)
                     return data
             except Exception as e: errors.append(f"تلاش ۴: {e}")
@@ -572,6 +594,9 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
     history = user_states[chat_id].get("history",[])
     new_parts =[]
     
+    is_image = True
+    has_non_image_file = False
+    
     if prompt: new_parts.append({"text": prompt})
     elif file_bytes: new_parts.append({"text": "لطفاً این فایل را به دقت بررسی کن."})
         
@@ -587,6 +612,11 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
             elif file_name.endswith(('.ogg', '.oga')): mime_type = "audio/ogg"
             elif file_name.endswith('.wav'): mime_type = "audio/wav"
             else: mime_type = "image/jpeg"
+            
+        is_image = mime_type.startswith('image/')
+        if not is_image:
+            has_non_image_file = True
+            
         new_parts.append({"inlineData": {"mimeType": mime_type, "data": base64_data}})
 
     if history and history[-1]["role"] == "user": history[-1]["parts"].extend(new_parts)
@@ -613,43 +643,60 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
                         except (KeyError, IndexError): continue 
             except Exception: continue
 
-    if not final_answer and HF_TOKENS:
-        hf_messages =[]
-        for msg in history:
-            role = "assistant" if msg["role"] == "model" else "user"
-            content_parts =[]
-            for part in msg["parts"]:
-                if "text" in part:
-                    content_parts.append({"type": "text", "text": part["text"]})
-                elif "inlineData" in part:
-                    mime_t = part["inlineData"]["mimeType"]
-                    b64_d = part["inlineData"]["data"]
-                    content_parts.append({
-                        "type": "image_url",
-                        "image_url": {"url": f"data:{mime_t};base64,{b64_d}"}
-                    })
-            hf_messages.append({"role": role, "content": content_parts})
-
-        keys_to_try_hf = HF_TOKENS.copy()
-        random.shuffle(keys_to_try_hf)
-        
-        async with aiohttp.ClientSession() as session:
-            for hf_key in keys_to_try_hf:
-                url = "https://router.huggingface.co/v1/chat/completions"
-                headers = {"Authorization": f"Bearer {hf_key}", "Content-Type": "application/json"}
-                payload = {
-                    "model": "google/gemma-4-31B-it:novita",
-                    "messages": hf_messages,
-                    "max_tokens": 4096
-                }
-                try:
-                    async with session.post(url, headers=headers, json=payload, timeout=60) as response:
-                        if response.status == 200:
-                            data = await response.json()
-                            final_answer = data["choices"][0]["message"]["content"]
-                            break
-                except Exception:
-                    continue
+    # اگر جیمینای جواب نداد
+    if not final_answer:
+        # اگر فایل ارسال شده عکس نبود، خطای اختصاصی می‌دهیم
+        if has_non_image_file:
+            if history and history[-1]["role"] == "user": history.pop()
+            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
+            await send_with_keyboard(client, chat_id, "❌ متأسفانه سرور اصلی در حال حاضر شلوغ است و در این شرایط **فعلاً تنها امکان تحلیل تصاویر** توسط سرور جایگزین در دسترس است.", False)
+            return
+            
+        # اگر فایل عکس بود یا فقط متن بود، می‌رویم سراغ مدل جایگزین (هاگینگ فیس)
+        if HF_TOKENS:
+            hf_messages =[]
+            for msg in history:
+                role = "assistant" if msg["role"] == "model" else "user"
+                content_parts =[]
+                for part in msg["parts"]:
+                    if "text" in part:
+                        content_parts.append({"type": "text", "text": part["text"]})
+                    elif "inlineData" in part:
+                        mime_t = part["inlineData"]["mimeType"]
+                        b64_d = part["inlineData"]["data"]
+                        if mime_t.startswith('image/'):
+                            content_parts.append({
+                                "type": "image_url",
+                                "image_url": {"url": f"data:{mime_t};base64,{b64_d}"}
+                            })
+                if content_parts:
+                    hf_messages.append({"role": role, "content": content_parts})
+
+            keys_to_try_hf = HF_TOKENS.copy()
+            random.shuffle(keys_to_try_hf)
+            
+            async with aiohttp.ClientSession() as session:
+                for hf_key in keys_to_try_hf:
+                    url = "https://router.huggingface.co/v1/chat/completions"
+                    headers = {"Authorization": f"Bearer {hf_key}", "Content-Type": "application/json"}
+                    payload = {
+                        "model": "google/gemma-4-31B-it:novita",
+                        "messages": hf_messages,
+                        "max_tokens": 4096
+                    }
+                    try:
+                        async with session.post(url, headers=headers, json=payload, timeout=60) as response:
+                            if response.status == 200:
+                                data = await response.json()
+                                final_answer = data["choices"][0]["message"]["content"]
+                                break
+                    except Exception:
+                        continue
 
     try:
         if proc_msg:
@@ -660,7 +707,7 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
 
     if not final_answer:
         if history and history[-1]["role"] == "user": history.pop()
-        await send_with_keyboard(client, chat_id, "❌ سرور شلوغ است فعلا بعدا امتحان کنید.", False)
+        await send_with_keyboard(client, chat_id, "❌ تمامی سرورها شلوغ هستند. لطفاً بعداً امتحان کنید.", False)
         return
 
     if not creds.get("is_premium"):
@@ -693,7 +740,7 @@ 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):
+async def process_image(client, chat_id, prompt, size_choice="1"):
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
     creds = get_user_credits(str_chat_id)
     if creds["image"] <= 0:
@@ -747,8 +794,21 @@ async def process_image(client, chat_id, prompt):
             if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
     
+    # تنظیم ابعاد انتخابی کاربر
+    w, h = 1024, 1024
+    size_name = "مربع (1:1) ⬛"
+    if size_choice == "2":
+        w, h = 768, 1344
+        size_name = "عمودی (9:16) 📱"
+    elif size_choice == "3":
+        w, h = 1344, 768
+        size_name = "افقی (16:9) 🖥️"
+    elif size_choice == "4":
+        w, h = 1024, 768
+        size_name = "استاندارد (4:3) 📸"
+
     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)
+    proc_msg = await send_with_keyboard(client, chat_id, f"🎨 در حال طراحی عکس...\n\n📏 **ابعاد:** {size_name}\n📝 **پرامپت ساخته شده:**\n`{short_preview}`\n\n(ممکن است چند ثانیه زمان ببرد)", False)
 
     keys_to_try = HF_TOKENS.copy()
     random.shuffle(keys_to_try)
@@ -758,7 +818,12 @@ 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(enhanced_prompt, model="Tongyi-MAI/Z-Image-Turbo")
+            generated_image = await hf_client.text_to_image(
+                enhanced_prompt, 
+                model="Tongyi-MAI/Z-Image-Turbo",
+                width=w,
+                height=h
+            )
             break 
         except Exception as e:
             last_error_log = str(e)
@@ -778,10 +843,10 @@ async def process_image(client, chat_id, prompt):
 
     try:
         file_name = f"image_{uuid.uuid4().hex}.jpg"
-        rgb_im = generated_image.convert('RGB')
-        rgb_im.save(file_name, format="JPEG", quality=100)
+        # استفاده از ترد برای جلوگیری از مسدود شدن برنامه حین کار با PIL
+        await asyncio.to_thread(sync_save_image, generated_image, file_name)
         await asyncio.sleep(1)
-        caption_text = f"🎨 تصویر شما با پرامپت هوشمند آماده شد!\n\n✨ ایده اولیه: {prompt}"
+        caption_text = f"🎨 تصویر شما با پرامپت هوشمند آماده شد!\n\n📏 ابعاد تصویر: {size_name}\n✨ ایده اولیه: {prompt}"
         upload_result = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
         if upload_result is not True: await send_with_keyboard(client, chat_id, f"❌ خطا در آپلود روبیکا:\n`{str(upload_result)[:800]}`", True)
         if os.path.exists(file_name): os.remove(file_name)
@@ -870,8 +935,8 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
 
     try:
         file_name = f"edited_flux_{uuid.uuid4().hex}.jpg"
-        rgb_im = generated_image.convert('RGB')
-        rgb_im.save(file_name, format="JPEG", quality=100)
+        # استفاده از ترد برای ذخیره عکس
+        await asyncio.to_thread(sync_save_image, generated_image, file_name)
         await asyncio.sleep(1)
         caption_text = f"🪄 ویرایش عکس با هوش مصنوعی Flux.2 انجام شد!\n\n✨ تغییرات خواسته شده: {prompt}\n🔤 متن ارسال شده به هوش: {translated_prompt}"
         upload_result = await helper_upload_file(client, chat_id, file_name, "Image", caption_text)
@@ -924,7 +989,8 @@ async def process_tts(client, chat_id, user_text, speaker_id, speaker_name):
             
             file_name_mp3 = f"audio_{uuid.uuid4().hex}.mp3"
             
-            with open(file_name_mp3, "wb") as f: f.write(audio_bytes)
+            # ذخیره فایل در ترد جداگانه
+            await asyncio.to_thread(sync_write_file, file_name_mp3, audio_bytes)
             
             await asyncio.sleep(1)
             upload_result_file = await helper_upload_file(client, chat_id, file_name_mp3, "File", "✅ صدای شما با موفقیت آماده شد (فایل MP3):")
@@ -1002,8 +1068,8 @@ async def process_podcast(client, chat_id, prompt):
             if not chunk_audio_bytes: return await send_with_keyboard(client, chat_id, f"❌ خطا در تولید صدای بخش {index+1} از سرور. عملیات متوقف شد.", True)
                 
             try:
-                audio_segment = AudioSegment.from_file(io.BytesIO(chunk_audio_bytes))
-                combined_audio += audio_segment
+                # جلوگیری از گیر کردن لوپ در هنگام ترکیب فایل‌های صوتی سنگین
+                combined_audio = await asyncio.to_thread(sync_combine_audio, combined_audio, chunk_audio_bytes)
             except Exception as e: return await send_with_keyboard(client, chat_id, f"❌ خطا در پردازش صدا (آیا pydub و ffmpeg نصب است؟):\n{str(e)}", True)
 
         if not creds.get("is_premium"):
@@ -1011,7 +1077,8 @@ async def process_podcast(client, chat_id, prompt):
             save_db(user_credits_db)
 
         file_name_mp3 = f"final_podcast_{uuid.uuid4().hex}.mp3"
-        combined_audio.export(file_name_mp3, format="mp3")
+        # اکسپورت فایل صوتی در ترد جدا
+        await asyncio.to_thread(sync_export_audio, combined_audio, file_name_mp3)
         
         try:
             if proc_msg:
@@ -1085,7 +1152,17 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
     proc_msg = await send_with_keyboard(client, chat_id, "👁️ در حال تحلیل فایل...", False)
     base64_data = base64.b64encode(file_bytes).decode('utf-8')
     mime_type, _ = mimetypes.guess_type(file_name)
-    if not mime_type: mime_type = "image/jpeg"
+    if not mime_type: 
+        if file_name.endswith(('.jpg', '.jpeg')): mime_type = "image/jpeg"
+        elif file_name.endswith('.png'): mime_type = "image/png"
+        elif file_name.endswith('.pdf'): mime_type = "application/pdf"
+        elif file_name.endswith('.mp4'): mime_type = "video/mp4"
+        elif file_name.endswith('.mp3'): mime_type = "audio/mp3"
+        elif file_name.endswith(('.ogg', '.oga')): mime_type = "audio/ogg"
+        elif file_name.endswith('.wav'): mime_type = "audio/wav"
+        else: mime_type = "image/jpeg"
+        
+    is_image = mime_type.startswith('image/')
     
     keys_to_try = get_next_gemini_keys(100)
     final_answer = None
@@ -1102,6 +1179,43 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
                         break
             except Exception: continue
 
+    # جایگزین تصویر اگر جیمینای جواب نداد
+    if not final_answer:
+        if is_image and HF_TOKENS:
+            keys_to_try_hf = HF_TOKENS.copy()
+            random.shuffle(keys_to_try_hf)
+            
+            hf_messages =[
+                {
+                    "role": "user",
+                    "content":[
+                        {"type": "text", "text": prompt},
+                        {
+                            "type": "image_url",
+                            "image_url": {"url": f"data:{mime_type};base64,{base64_data}"}
+                        }
+                    ]
+                }
+            ]
+            
+            async with aiohttp.ClientSession() as session:
+                for hf_key in keys_to_try_hf:
+                    url = "https://router.huggingface.co/v1/chat/completions"
+                    headers = {"Authorization": f"Bearer {hf_key}", "Content-Type": "application/json"}
+                    payload = {
+                        "model": "google/gemma-4-31B-it:novita",
+                        "messages": hf_messages,
+                        "max_tokens": 4096
+                    }
+                    try:
+                        async with session.post(url, headers=headers, json=payload, timeout=60) as response:
+                            if response.status == 200:
+                                data = await response.json()
+                                final_answer = data["choices"][0]["message"]["content"]
+                                break
+                    except Exception:
+                        continue
+
     try:
         if proc_msg:
             msg_id = getattr(proc_msg, 'message_id', None)
@@ -1115,7 +1229,10 @@ async def process_file_analysis(client, chat_id, file_bytes, file_name, prompt):
             save_db(user_credits_db)
         await send_with_keyboard(client, chat_id, f"💡 **نتیجه تحلیل:**\n\n{final_answer}", True)
     else: 
-        await send_with_keyboard(client, chat_id, "❌ سرور شلوغ است فعلا بعدا امتحان کنید.", True)
+        if not is_image:
+            await send_with_keyboard(client, chat_id, "❌ متأسفانه سرور اصلی در حال حاضر شلوغ است و در این شرایط **فعلاً تنها امکان تحلیل تصاویر** توسط سرور جایگزین در دسترس است.", True)
+        else:
+            await send_with_keyboard(client, chat_id, "❌ تمامی سرورها شلوغ هستند. لطفاً بعداً امتحان کنید.", True)
 
 
 async def process_create_file(client, chat_id, topic):
@@ -1136,7 +1253,7 @@ async def process_create_file(client, chat_id, topic):
     async with aiohttp.ClientSession() as session:
         for key in keys_to_try:
             url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-2.5-flash:generateContent?key={key}"
-            payload = {"contents": [{"parts":[{"text": ai_prompt}]}], "generationConfig": {"temperature": 0.7, "maxOutputTokens": 8192}}
+            payload = {"contents":[{"parts":[{"text": ai_prompt}]}], "generationConfig": {"temperature": 0.7, "maxOutputTokens": 8192}}
             try:
                 async with session.post(url, json=payload, timeout=60) as response:
                     if response.status == 200:
@@ -1175,31 +1292,37 @@ async def process_create_file(client, chat_id, topic):
             if msg_id: await client.delete_messages(chat_id,[msg_id])
     except Exception: pass
 
-    proc_msg = await send_with_keyboard(client, chat_id, "📄 مقاله با موفقیت نوشته شد! در حال ارتباط با سرور و تبدیل به فایل‌های PDF و Word...\n(لطفا صبور باشید)", False)
-
-    pdf_bytes = None
-    docx_bytes = None
     converter_url = "https://opera8-texttopdf.hf.space/"
+    uid = uuid.uuid4().hex
+    
+    # پیام اطلاع رسانی قبل از ساخت فایل‌ها
+    proc_msg = await send_with_keyboard(client, chat_id, "📄 مقاله با موفقیت نوشته شد!\n\n⏳ در حال ارتباط همزمان با هوش مصنوعی برای دریافت فایل‌های PDF و Word...\n", False)
 
-    async with aiohttp.ClientSession() as session:
-        try:
-            form_data_pdf = aiohttp.FormData()
-            form_data_pdf.add_field('content', article_text)
-            form_data_pdf.add_field('format', 'pdf')
-            async with session.post(converter_url, data=form_data_pdf, timeout=60) as resp:
-                if resp.status == 200:
-                    pdf_bytes = await resp.read()
-        except Exception as e: print("PDF creation error:", e)
-
-        try:
-            form_data_docx = aiohttp.FormData()
-            form_data_docx.add_field('content', article_text)
-            form_data_docx.add_field('format', 'docx')
-            async with session.post(converter_url, data=form_data_docx, timeout=60) as resp:
-                if resp.status == 200:
-                    docx_bytes = await resp.read()
-        except Exception as e: print("DOCX creation error:", e)
-
+    # تابع کمکی برای درخواست همزمان به سرور
+    async def fetch_document(fmt):
+        for attempt in range(3):
+            try:
+                async with aiohttp.ClientSession() as session:
+                    form_data = aiohttp.FormData()
+                    form_data.add_field('content', article_text)
+                    form_data.add_field('format', fmt)
+                    # تایم‌اوت 180 ثانیه برای فرصت کافی به Weasyprint
+                    async with session.post(converter_url, data=form_data, timeout=180) as resp:
+                        if resp.status == 200:
+                            data = await resp.read()
+                            if data and len(data) > 0:
+                                return data
+            except Exception:
+                pass
+            await asyncio.sleep(2)
+        return None
+
+    # اجرای همزمان هر دو درخواست برای سرعت بیشتر و جلوگیری از تایم‌اوت متوالی
+    pdf_bytes, docx_bytes = await asyncio.gather(
+        fetch_document('pdf'),
+        fetch_document('docx')
+    )
+    
     try:
         if proc_msg:
             msg_id = getattr(proc_msg, 'message_id', None)
@@ -1207,38 +1330,40 @@ async def process_create_file(client, chat_id, topic):
             if msg_id: await client.delete_messages(chat_id, [msg_id])
     except Exception: pass
 
-    if not pdf_bytes and not docx_bytes:
-        return await send_with_keyboard(client, chat_id, "❌ متاسفانه در تبدیل متن به فایل خطایی رخ داد.", True)
-
-    if not creds.get("is_premium"):
-        user_credits_db[str_chat_id]["chat"] -= 1
-        save_db(user_credits_db)
-
-    uid = uuid.uuid4().hex
-    pdf_filename = f"Article_{uid}.pdf"
-    docx_filename = f"Article_{uid}.docx"
-
-    uploaded_any = False
+    # --- آپلود فایل‌ها در روبیکا ---
+    uploaded_pdf = False
+    uploaded_docx = False
     
     if pdf_bytes:
-        with open(pdf_filename, "wb") as f:
-            f.write(pdf_bytes)
+        pdf_filename = f"Article_{uid}.pdf"
+        await asyncio.to_thread(sync_write_file, pdf_filename, pdf_bytes)
         res = await helper_upload_file(client, chat_id, pdf_filename, "File", f"📄 فایل PDF مقاله شما:\n\n💡 موضوع: {topic}")
-        if res is True: uploaded_any = True
-        if os.path.exists(pdf_filename): os.remove(pdf_filename)
-        await asyncio.sleep(1.5)
+        if res is True: 
+            uploaded_pdf = True
+        if os.path.exists(pdf_filename): 
+            os.remove(pdf_filename)
+            
+    await asyncio.sleep(1.5)
 
     if docx_bytes:
-        with open(docx_filename, "wb") as f:
-            f.write(docx_bytes)
+        docx_filename = f"Article_{uid}.docx"
+        await asyncio.to_thread(sync_write_file, docx_filename, docx_bytes)
         res = await helper_upload_file(client, chat_id, docx_filename, "File", f"📝 فایل Word (DOCX) مقاله ��ما:\n\n💡 موضوع: {topic}")
-        if res is True: uploaded_any = True
-        if os.path.exists(docx_filename): os.remove(docx_filename)
+        if res is True: 
+            uploaded_docx = True
+        if os.path.exists(docx_filename): 
+            os.remove(docx_filename)
 
-    if uploaded_any:
-        await send_with_keyboard(client, chat_id, "✅ مقاله شما با موفقیت به صورت فایل تحویل داده شد!", True)
+    if not creds.get("is_premium"):
+        user_credits_db[str_chat_id]["chat"] -= 1
+        save_db(user_credits_db)
+
+    if uploaded_pdf and uploaded_docx:
+        await send_with_keyboard(client, chat_id, "✅ هر دو فایل (PDF و Word) مقاله شما با موفقیت تحویل داده شد!", True)
+    elif uploaded_pdf or uploaded_docx:
+        await send_with_keyboard(client, chat_id, "⚠️ یکی از فایل‌ها با موفقیت ارسال شد، اما در ساخت یا ارسال فایل دیگر مشکلی از سمت سرور پیش آمد.", True)
     else:
-        await send_with_keyboard(client, chat_id, "❌ فایل‌ها ساخته شدند اما روبیکا در ارسال آن‌ها دچار مشکل شد.", True)
+        await send_with_keyboard(client, chat_id, "❌ فایل‌ها ساخته شدند اما متأسفانه روبیکا در ارسال آن‌ها دچار مشکل شد.", True)
 
 
 if not bot_token:
@@ -1382,7 +1507,7 @@ else:
                                     expire_txt = f"{jy:04d}/{jm:02d}/{jd:02d}"
                                 status_text = f"🌟 نسخه پرو (ویژه)\n📅 انقضا: {expire_txt}\n⏳ باقیمانده: {days_left} روز"
                             else:
-                                status_text = "🥉 نسخه رایگان (آزمایشی)\n⏳ سهمیه روزانه"
+                                status_text = "🥉 نسخه رایگان (آزمایشی)\n⏳ وضعیت: سهمیه روزانه محدود"
 
                             chat_rem = "نامحدود ∞" if is_prem else t_creds.get('chat', 0)
                             podcast_rem = "نامحدود ∞" if is_prem else t_creds.get('podcast', 0)
@@ -1392,23 +1517,27 @@ else:
                             image_rem = t_creds.get('image', 0)
                             edit_image_rem = t_creds.get('edit_image', 0)
                             invs = t_creds.get('invited_count', 0)
+                            joined_status = "بله ✅" if t_creds.get('has_joined') else "خیر ❌"
 
-                            info_msg = f"""🔍 **اطلاعات کاربر `{target_id}`:**
+                            info_msg = f"""🔍 **اطلاعات اختصاصی کاربر:**
+👤 شناسه: `{target_id}`
+عضو کانال: {joined_status}
 
-🔹 **وضعیت:** {status_text}
+🔹 **وضعیت اشتراک:**
+{status_text}
 🎁 **تعداد افراد دعوت شده:** {invs} نفر
 
-📊 **سهمیه:**
-- چت: {chat_rem}
-- تولید عکس: {image_rem}
-- ویرایش عکس: {edit_image_rem}
-- پادکست: {podcast_rem}
-- متن به صدا: {tts_rem}
-- تحلیل فایل: {file_rem}
-- صدا به متن: {stt_rem}"""
+📊 **سهمیه باقی‌مانده فعلی:**
+💬 چت: {chat_rem}
+🎨 تولید عکس: {image_rem}
+🪄 ویرایش عکس: {edit_image_rem}
+🎙 پادکست: {podcast_rem}
+🗣 متن به صدا: {tts_rem}
+📁 تحلیل فایل: {file_rem}
+📝 صدا به متن: {stt_rem}"""
                             await send_with_keyboard(client, chat_id, info_msg, False)
                         else:
-                            await send_with_keyboard(client, chat_id, f"❌ کاربری با شناسه `{target_id}` در دیتابیس یافت نشد.", False)
+                            await send_with_keyboard(client, chat_id, f"❌ کاربری با شناسه `{target_id}` در دیتابیس ربات یافت نشد.", False)
                 return
 
             is_file = False
@@ -1594,7 +1723,7 @@ else:
 🔑 **شناسه یکتای شما:** `{chat_id}`
 
 👨‍💻 **ارتباط با پشتیبانی:**
-🆔 @H_a_m_e_d100"""
+🆔 @aialpha_admin"""
                 await send_with_keyboard(client, chat_id, buy_text, True)
                 return
 
@@ -1608,7 +1737,7 @@ else:
 🔑 **شناسه یکتای ربات شما:** `{chat_id}`
 
 👨‍💻 **دقت کنید شناسه ربات رو به پشتیبانی داخل خود برنامه هوش مصنوعی آلفا ارسال کنید:**
-🆔 @H_a_m_e_d100"""
+🤖"""
                 await send_with_keyboard(client, chat_id, transfer_text, True)
                 return
 
@@ -1620,7 +1749,7 @@ else:
 
             if user_text_str in["/image", "🎨 عکس", "ساخت تصاویر🎨"]:
                 user_states[str_chat_id]["mode"] = "image_waiting_for_text"
-                await send_with_keyboard(client, chat_id, "🎨 شما وارد بخش **ساخت عکس پیشرفته** شدید.\n\nمتن خود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                await send_with_keyboard(client, chat_id, "🎨 شما وارد بخش **ساخت عکس پیشرفته** شدید.\n\nمتن (ایده) خود را به صورت کامل ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
             if user_text_str in["/edit_image", "ویرایش تصاویر 🪄"]:
@@ -1745,8 +1874,35 @@ else:
                 elif user_text_str: asyncio.create_task(process_gemini(client, chat_id, user_text_str))
                 return
 
+            # ===============================================
+            # آپدیت جدید: دریافت سایز قبل از ساخت عکس
+            # ===============================================
             elif current_mode == "image_waiting_for_text":
-                if user_text_str: asyncio.create_task(process_image(client, chat_id, user_text_str))
+                if user_text_str:
+                    user_states[str_chat_id]["text"] = user_text_str
+                    user_states[str_chat_id]["mode"] = "image_waiting_for_size"
+                    
+                    size_menu = """🖼 **لطفاً ابعاد تصویر خود را انتخاب کنید:**
+
+1️⃣ `1:1` (مربع) - مناسب پروفایل ⬛
+2️⃣ `9:16` (عمودی) - مناسب استوری و پس‌زمینه گوشی 📱
+3️⃣ `16:9` (افقی) - مناسب دسکتاپ و ویدیو 🖥️
+4️⃣ `4:3` (استاندارد) - مناسب چاپ و پست 📸
+
+(لطفاً فقط عدد 1 تا 4 را بفرستید)"""
+                    await send_with_keyboard(client, chat_id, size_menu, False)
+                else:
+                    await send_with_keyboard(client, chat_id, "⚠️ لطفاً ایده عکس خود را به صورت متنی بفرستید.", False)
+                return
+
+            elif current_mode == "image_waiting_for_size":
+                normalized_choice = to_english_digits(user_text_str).strip()
+                if normalized_choice in["1", "2", "3", "4"]:
+                    saved_text = user_states[str_chat_id].get("text", "")
+                    user_states[str_chat_id]["mode"] = None
+                    asyncio.create_task(process_image(client, chat_id, saved_text, normalized_choice))
+                else:
+                    await send_with_keyboard(client, chat_id, "❌ عدد وارد شده نامعتبر است! لطفاً فقط یکی از اعداد 1 تا 4 را بفرستید.", False)
                 return
 
             elif current_mode == "image_edit_waiting_for_image":
@@ -1824,7 +1980,7 @@ else:
                     user_states[str_chat_id]["file_bytes"] = file_bytes
                     user_states[str_chat_id]["file_name"] = file_name 
                     user_states[str_chat_id]["mode"] = "file_waiting_for_prompt"
-                    await send_with_keyboard(client, chat_id, "✅ فایل با موفقیت دریافت شد.\n\nحالا لطفاً متنی بگویید **چگونه تحلیل شود؟**", False)
+                    await send_with_keyboard(client, chat_id, "✅ فایل با موفقیت دریافت شد.\n\nحالا لطفاً متنی بگویید **چگونه تحلیل شود؟**\n(می‌توانید سوال خاصی بپرسید یا فقط بخواهید فایل را توضیح دهد)", False)
                 except Exception as dl_err: await send_with_keyboard(client, chat_id, f"❌ خطا در دریافت فایل!\n{str(dl_err)}", False)
                 return
 
@@ -1848,6 +2004,12 @@ else:
 
 if __name__ == "__main__":
     threading.Thread(target=run_flask, daemon=True).start()
+    
     if bot_token:
-        print("ربات آلفا پرو با سیستم اشتراک نامحدود + کد هدیه یکتا + سپر امنیتی روشن شد...")
+        # افزایش ظرفیت کارگرهای (Workers) پس‌زمینه در Asyncio برای افزایش شدید سرعت پاسخگویی
+        loop = asyncio.get_event_loop()
+        # تنظیم ترد پول روی 16 کارگر به جای 1 پیش‌فرض که باعث کندی میشد
+        loop.set_default_executor(concurrent.futures.ThreadPoolExecutor(max_workers=16))
+        
+        print("ربات آلفا پرو با سیستم اشتراک نامحدود + سپر امنیتی + 16 Worker پس‌زمینه روشن شد...")
         bot.run()