Update main.py

main.py

@@ -1334,43 +1334,50 @@ async def process_image(client, chat_id, prompt, size_choice="1"):
     if not HF_TOKENS: return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
     
     proc_msg = await send_with_keyboard(client, chat_id, "✨ در حال ترجمه و بهینه‌سازی پرامپت شما...\n(تبدیل به پرامپت حرفه‌ای)", False)
-    enhanced_prompt = prompt 
-    gemini_sys_prompt = f"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. Return ONLY the final English prompt string.\nUser input: {prompt}"
     
-    if GEMINI_KEYS:
-        keys_to_try_gemini = get_next_gemini_keys(100)
-        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
+    aya_sys_prompt = f"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. Return ONLY the final English prompt string.\nUser input: {prompt}"
+    
+    enhanced_prompt = prompt
+    session_hash = ''.join(random.choices(string.ascii_lowercase + string.digits, k=11))
+    join_url = "https://coherelabs-aya-expanse.hf.space/gradio_api/queue/join"
+    data_url = f"https://coherelabs-aya-expanse.hf.space/gradio_api/queue/data?session_hash={session_hash}"
+    
+    payload = {
+        "data": [
+            aya_sys_prompt,
+            [],
+            None,
+            None
+        ],
+        "event_data": None,
+        "fn_index": 2,
+        "session_hash": session_hash,
+        "trigger_id": 37
+    }
 
-    if enhanced_prompt == prompt and HF_TOKENS:
-        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":[{"role": "user", "content":[{"type": "text", "text": gemini_sys_prompt}]}],
-                    "max_tokens": 1024
-                }
-                try:
-                    async with session.post(url, headers=headers, json=payload, timeout=20) as response:
-                        if response.status == 200:
-                            data = await response.json()
-                            enhanced_prompt = data["choices"][0]["message"]["content"].strip()
-                            break
-                except Exception:
-                    continue
+    for attempt in range(3):
+        try:
+            async with aiohttp.ClientSession() as session:
+                async with session.post(join_url, json=payload, timeout=30) as resp:
+                    if resp.status == 200:
+                        async with session.get(data_url, timeout=120) as data_resp:
+                            async for line_bytes in data_resp.content:
+                                line = line_bytes.decode('utf-8').strip()
+                                if line.startswith("data: "):
+                                    try:
+                                        json_data = json.loads(line[6:])
+                                        if json_data.get("msg") == "process_completed":
+                                            if json_data.get("success"):
+                                                output_data = json_data.get("output", {}).get("data", [])
+                                                if output_data and len(output_data) > 0:
+                                                    result_hist = output_data[0]
+                                                    if isinstance(result_hist, list) and len(result_hist) > 0:
+                                                        enhanced_prompt = result_hist[-1][1].strip()
+                                            break
+                                    except Exception: pass
+        except Exception: pass
+        if enhanced_prompt != prompt: break
+        await asyncio.sleep(2)
 
     try:
         if proc_msg:
@@ -1953,51 +1960,51 @@ async def process_create_file(client, chat_id, topic):
     if creds["chat"] <= 0:
         return await send_with_keyboard(client, chat_id, "❌ اعتبار چت شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
 
-    if not GEMINI_KEYS: return await send_with_keyboard(client, chat_id, "❌ کلیدهای جیمینای تنظیم نشده‌اند.", False)
-    
     proc_msg = await send_with_keyboard(client, chat_id, "✍️ در حال تحقیق و نگارش مقاله جامع و حرفه‌ای...\n(این عملیات با توجه به طولانی بودن متن ممکن است کمی طول بکشد)", False)
 
-    ai_prompt = f"یک مقاله بسیار جامع، کاملا حرفه‌ای، بسیار طولانی و با جزئیات کامل درباره موضوع زیر به زبان فارسی بنویس. فقط متن مقاله را بده و هیچ توضیح اضافه‌ای نده:\n\nموضوع: {topic}"
+    ai_prompt = f"یک مقاله بی‌نهایت جامع، کاملاً حرفه‌ای و بسیار بسیار طولانی درباره موضوع زیر به زبان فارسی بنویس. دقت کن که مقاله باید شامل بخش‌بندی‌های متعدد باشد و توضیحات زیر هر عنوان باید به شدت طولانی، مفصل و با جزئیات و مثال‌های فراوان (حداقل چند پاراگراف بلند برای هر بخش) نوشته شود؛ به هیچ وجه زیر عناوین توضیحات کوتاه نده. فقط متن اصلی مقاله را بده و هیچ توضیح اضافه‌ای ننویس:\n\nموضوع: {topic}"
     
     article_text = None
-    for attempt in range(5):
-        keys_to_try = get_next_gemini_keys(100)
-        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}}
-                try:
-                    async with session.post(url, json=payload, timeout=60) as response:
-                        if response.status == 200:
-                            data = await response.json()
-                            article_text = data["candidates"][0]["content"]["parts"][0]["text"]
-                            break
-                except Exception: continue
-        if article_text: break
-        await asyncio.sleep(2)
+    session_hash = ''.join(random.choices(string.ascii_lowercase + string.digits, k=11))
+    join_url = "https://coherelabs-aya-expanse.hf.space/gradio_api/queue/join"
+    data_url = f"https://coherelabs-aya-expanse.hf.space/gradio_api/queue/data?session_hash={session_hash}"
+    
+    payload = {
+        "data": [
+            ai_prompt,
+            [],
+            None,
+            None
+        ],
+        "event_data": None,
+        "fn_index": 2,
+        "session_hash": session_hash,
+        "trigger_id": 37
+    }
 
-    if not article_text and HF_TOKENS:
-        for attempt in range(5):
-            keys_to_try_hf = HF_TOKENS.copy()
-            random.shuffle(keys_to_try_hf)
+    for attempt in range(3):
+        try:
             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":[{"role": "user", "content":[{"type": "text", "text": ai_prompt}]}],
-                        "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()
-                                article_text = data["choices"][0]["message"]["content"]
-                                break
-                    except Exception: continue
-            if article_text: break
-            await asyncio.sleep(2)
+                async with session.post(join_url, json=payload, timeout=30) as resp:
+                    if resp.status == 200:
+                        async with session.get(data_url, timeout=180) as data_resp:
+                            async for line_bytes in data_resp.content:
+                                line = line_bytes.decode('utf-8').strip()
+                                if line.startswith("data: "):
+                                    try:
+                                        json_data = json.loads(line[6:])
+                                        if json_data.get("msg") == "process_completed":
+                                            if json_data.get("success"):
+                                                output_data = json_data.get("output", {}).get("data", [])
+                                                if output_data and len(output_data) > 0:
+                                                    result_hist = output_data[0]
+                                                    if isinstance(result_hist, list) and len(result_hist) > 0:
+                                                        article_text = result_hist[-1][1]
+                                            break
+                                    except Exception: pass
+        except Exception: pass
+        if article_text: break
+        await asyncio.sleep(2)
 
     if not article_text:
         return await send_with_keyboard(client, chat_id, "❌ سرور تولید متن شلوغ است، لطفاً بعداً تلاش کنید.", True)
@@ -2009,7 +2016,7 @@ 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)
+    proc_msg = await send_with_keyboard(client, chat_id, "📄 مقاله با موفقیت نوشته شد! در حال ارتباط با هوش مصنوعی و دریافت فایل‌های PDF و Word...\n(لطفا صبور باشید)", False)
 
     converter_url = "https://opera8-texttopdf.hf.space/"
     uid = uuid.uuid4().hex
@@ -2597,14 +2604,45 @@ else:
                 await send_with_keyboard(client, chat_id, "🪄 به بخش **ویرایش عکس (Flux.2)** خوش آمدید.\n\nلطفاً ابتدا عکسی که می‌خواهید ویرایش کنید را بفرستید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
-            if user_text_str in["/tts", "🎙️ صدا", "تبدیل متن به صدا🗣️"]:
+            if user_text_str in ["/tts", "🎙️ صدا", "تبدیل متن به صدا🗣️"]:
                 user_states[str_chat_id]["mode"] = "tts_waiting_for_text"
-                await send_with_keyboard(client, chat_id, "🎙️ شما وارد بخش **تبدیل متن به صدا** شدید.\n\nلطفاً متنی که می‌خواهید به صدا تبدیل شود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                tts_msg = """🎙️ شما وارد بخش **تبدیل متن به صدا** شدید.
+
+
+📝 لطفاً متنی که می‌خواهید به صدا تبدیل شود را ارسال کنید.
+
+🌍 **پشتیبانی از تمامی زبان‌های دنیا!**
+🎭 **درک لحن و احساسات:**
+هوش مصنوعی ما احساسات رو درک می‌کنه! کافیه لحن دلخواهت رو داخل پرانتز بنویسی.
+💡 *مثال:* هی، گوش کن! یه راز دارم که فقط به تو می‌گم. (با لحنی مرموز، شیطنت‌آمیز و نجواگونه)
+
+(برای خروج دکمه «برگشت♻️» را بزنید)"""
+                await send_with_keyboard(client, chat_id, tts_msg, True)
                 return
             
-            if user_text_str in["/podcast", "📻 پادکست", "ساخت پادکست 🎙️"]:
+            if user_text_str in ["/podcast", "📻 پادکست", "ساخت پادکست 🎙️"]:
                 user_states[str_chat_id]["mode"] = "podcast_waiting_for_topic"
-                await send_with_keyboard(client, chat_id, "📻 شما وارد بخش **ساخت پادکست** شدید.\n\nلطفاً موضوع پادکست خود را بفرستید.\nمثال: درباره تاریخچه پیدایش قهوه با ۳ گوینده یک پادکست جذاب بساز . همچنین این قسمت متصل به مدل زبانی است و درخواست هارو قبل از ساخت درک میکنه. میتوانید مقاله کامل یک سایت بفرستید با تبلیغات یا هرچی، هوش مصنوعی متن مقاله رو استخراج و پادکست براتون میسازه . در توضیحات امکان مشخص کردن تعداد گوینده به همراه اسم شون نیز از سمت شما امکان پذیر است.\n\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                podcast_msg = """📻 شما وارد بخش **ساخت پادکست** شدید.
+
+لطفاً موضوع پادکست خود را بفرستید. این بخش متصل به هوش مصنوعی زبانی است و درخواست شما را قبل از ساخت کاملاً درک می‌کند! 🧠
+
+➖➖➖➖➖➖➖➖➖➖
+💡 **راهنم��ی استفاده:**
+🔹 می‌توانید فقط یک موضوع بدهید (مثال: درباره تاریخچه پیدایش قهوه با سه گوینده یک پادکست جذاب بساز).
+🔹 می‌توانید متن کامل یک مقاله از یک سایت را بفرستید تا هوش مصنوعی خودش محتوای اصلی را استخراج کرده و پادکست بسازد.
+🔹 امکان مشخص کردن تعداد گویندگان و اسم آن‌ها به انتخاب شما وجود دارد!
+
+➖➖➖➖➖➖➖➖➖➖
+🎙️ **لیست ۳۰ گوینده اختصاصی ما:**
+شهاب | آوا | نوید | آرمان | مهسا | دانا | سامان | آرش | شبنم | سحر | مریم | بهرام | نیکان | فرناز | سارا | مانی | آرتین | دلنواز | روژان | امید | بردیا | ترانه | نیکو | هستی | کامیار | کیانوش | پویا | مهتاب | سام | لیدا
+
+➖➖➖➖➖➖➖➖➖➖
+📌 **مثال برای درخواست:**
+یک پادکست برام بساز در باره مدیریت زمان با دو گوینده امید و مهسا
+اگر گوینده و تعداد آن از سمت شما مشخص نشه هوش مصنوعی خودش انتخاب میکنه
+
+(برای خروج دکمه «برگشت♻️» را بزنید)"""
+                await send_with_keyboard(client, chat_id, podcast_msg, True)
                 return
 
             if user_text_str in["/vc", "تغییر صدا 🎙️"]:
@@ -2617,7 +2655,7 @@ else:
                 user_states[str_chat_id]["mode"] = "clone_waiting_for_src"
                 user_states[str_chat_id]["file_bytes"] = None
                 user_states[str_chat_id]["ref_bytes"] = None
-                clone_text = """👤 **شما وارد بخش کلون کردن صدای اختصاصی شدید.**
+                clone_text = """👤 **شما وارد بخش کلون کردن صدای اختصاصی شدید. با یک فایل صوتی کوتاه از هر شخصی صدای خودتونو به همون صدا تغییر بدید.**
 
 ✨ **این بخش چه کاری انجام می‌دهد؟**
 با این ویژگی بی‌نظیر، شما می‌توانید صدای خود (یا هر فرد دیگری) را دقیقاً شبیه‌سازی کنید! کافیست یک نمونه از صدای خودتان صحبت کنید و یک نمونه از صدای شخص مورد نظر (الگو) را به هوش مصنوعی بدهید. ربات لحن و کلمات شما را گرفته و دقیقاً با جنس صدای شخص الگو بازسازی می‌کند.