Update main.py

main.py

@@ -10,6 +10,7 @@ import mimetypes
 import io
 import json
 import datetime
+import string
 from flask import Flask
 from rubpy.bot import BotClient, filters
 
@@ -573,6 +574,41 @@ async def process_image(client, chat_id, prompt):
         if os.path.exists(file_name): os.remove(file_name)
     except Exception as e: await send_with_keyboard(client, chat_id, f"❌ خطا در ذخیره عکس:\n{str(e)[:150]}", True)
 
+# --- تابع جدید برای ارسال متن به اسپیس مترجم ---
+async def translate_text_aloha(prompt_text):
+    session_hash = ''.join(random.choices(string.ascii_lowercase + string.digits, k=11))
+    join_url = "https://hamed744-translate-tts-aloha.hf.space/gradio_api/queue/join"
+    payload = {
+        "data": [prompt_text, "انگلیسی (آمریکا) - جنی (زن)", 0, 0, 0],
+        "fn_index": 1,
+        "session_hash": session_hash
+    }
+    
+    try:
+        async with aiohttp.ClientSession() as session:
+            # پیوستن به صف ترجمه
+            async with session.post(join_url, json=payload, timeout=20) as resp:
+                if resp.status != 200:
+                    return prompt_text
+                    
+            # دریافت داده‌ها به صورت جریانی (Stream)
+            data_url = f"https://hamed744-translate-tts-aloha.hf.space/gradio_api/queue/data?session_hash={session_hash}"
+            async with session.get(data_url, timeout=60) as resp:
+                async for line_bytes in 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"):
+                                    return json_data["output"]["data"][0]
+                                break
+                        except Exception:
+                            pass
+    except Exception:
+        pass
+        
+    return prompt_text
 
 # --- ۲.۵. پردازش ویرایش عکس با FLUX.2-dev ---
 async def process_image_edit(client, chat_id, image_bytes, prompt):
@@ -584,8 +620,13 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
     if not HF_TOKENS:
         return await send_with_keyboard(client, chat_id, "❌ توکن‌های هاگینگ فیس تنظیم نشده‌اند.", False)
 
-    proc_msg = await send_with_keyboard(client, chat_id, "🪄 در حال پردازش و اعمال جادوی FLUX.2 روی عکس شما...\n(این فرآیند ممکن است کمی طول بکشد)", False)
+    proc_msg = await send_with_keyboard(client, chat_id, "🪄 در حال ترجمه دستور شما توسط اسپیس و اعمال جادوی FLUX.2...\n(این فرآیند ممکن است کمی طول بکشد)", False)
     
+    # گرفتن ترجمه از اسپیس درخواستی
+    translated_prompt = await translate_text_aloha(prompt)
+    if not translated_prompt or translated_prompt.strip() == "":
+        translated_prompt = prompt
+
     keys_to_try = HF_TOKENS.copy()
     random.shuffle(keys_to_try)
     generated_image = None
@@ -596,7 +637,7 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
             hf_client = AsyncInferenceClient(provider="fal-ai", api_key=token)
             generated_image = await hf_client.image_to_image(
                 image_bytes,
-                prompt=prompt,
+                prompt=translated_prompt,
                 model="black-forest-labs/FLUX.2-dev"
             )
             break 
@@ -623,7 +664,7 @@ async def process_image_edit(client, chat_id, image_bytes, prompt):
         rgb_im = generated_image.convert('RGB')
         rgb_im.save(file_name, format="JPEG", quality=100)
         await asyncio.sleep(1)
-        caption_text = f"🪄 ویرایش عکس با هوش مصنوعی Flux.2 انجام شد!\n\n✨ تغییرات خواسته شده: {prompt}"
+        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)
         if upload_result is not True:
             await send_with_keyboard(client, chat_id, f"❌ عکس ساخته شد اما خطا در ارسال به روبیکا رخ داد:\n`{str(upload_result)[:800]}`", True)