Update main.py

main.py

@@ -1995,7 +1995,7 @@ async def process_podcast(client, chat_id, prompt):
             os.remove(file_name_mp3)
 
 # ==============================================================================
-# 🟢 پارت 18: استخراج متن از صدا (STT) و تحلیل تصویر (متصل به هوش مصنوعی Gemma و Aya)
+# 🟢 پارت 18: استخراج متن از صدا (STT) 
 # ==============================================================================
 async def process_stt(client, chat_id, audio_bytes, file_name):
     str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
@@ -2046,44 +2046,6 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     else: 
         await send_with_keyboard(client, chat_id, "❌ سرور شلوغ است فعلا بعدا امتحان کنید.", True)
 
-async def process_image_analysis(client, chat_id, file_bytes, file_name, prompt):
-    str_chat_id = str(chat_id).replace("`", "").replace("'", "").replace('"', "").strip()
-    creds = get_user_credits(str_chat_id)
-    # از همان سهمیه قبلی فایل برای تحلیل تصویر استفاده میکنیم تا در دیتابیس ارور نگیرید
-    if creds["file"] <= 0:
-        return await send_with_keyboard(client, chat_id, "❌ اعتبار تحلیل تصویر شما تمام شده است. لطفاً از منوی اصلی وارد بخش «خرید اشتراک 💎» شوید.", False)
-
-    proc_msg = await send_with_keyboard(client, chat_id, "👁️ در حال درک و تحلیل دقیق تصویر توسط هوش مصنوعی...", False)
-    
-    mime_type, _ = mimetypes.guess_type(file_name)
-    
-    if not mime_type: 
-        if file_name.endswith(('.jpg', '.jpeg')): mime_type = "image/jpeg"
-        elif file_name.endswith('.png'): mime_type = "image/png"
-        else: mime_type = "image/jpeg"
-        
-    gemma_prompt = prompt if prompt else "این تصویر را با جزئیات کامل تحلیل کن"
-    
-    # استفاده از سیستم هوشمند (تلاش ۵ باره در اکشن + انتقال نامرئی به مدل Aya)
-    final_answer = await smart_vision_analysis(gemma_prompt, file_bytes, file_name, mime_type)
-
-    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
-
-    if final_answer: 
-        # 🧹 اعمال فیلتر پاکسازی فرمول‌های ریاضی و لاتک
-        final_answer = clean_math_formatting(final_answer)
-        sent = await send_with_keyboard(client, chat_id, f"💡 **نتیجه تحلیل تصویر:**\n\n{final_answer}", True)
-        if sent and not creds.get("is_premium"):
-            user_credits_db[str_chat_id]["file"] -= 1
-            save_db(user_credits_db)
-    else: 
-        await send_with_keyboard(client, chat_id, "❌ سرور تحلیل تصویر شلوغ است یا خطایی رخ داد. لطفاً کمی بعد امتحان کنید.", True)
-
 # ==============================================================================
 # 🟢 پارت 19: ساخت مقاله و خروجی فایل متنی (Word/PDF)
 # ==============================================================================
@@ -2702,7 +2664,7 @@ else:
             if is_user_spamming(str_chat_id): return
 
             if str_chat_id not in user_states:
-                user_states[str_chat_id] = {"mode": None, "text": "", "history":[], "file_bytes": None, "file_name": None, "last_time": time.time(), "video_type": None, "selected_model": None, "vid_width": 1024, "vid_height": 1024}
+                user_states[str_chat_id] = {"mode": None, "text": "", "history":[], "file_bytes": None, "file_name": None, "last_time": time.time(), "video_type": None, "selected_model": None, "vid_width": 1024, "vid_height": 1024, "entry_point": None}
             else:
                 user_states[str_chat_id]["last_time"] = time.time()  
 
@@ -3091,10 +3053,18 @@ else:
 
             if user_text_str in["/chat", "💬 چت", "چت با هوش مصنوعی 🤖"]:
                 user_states[str_chat_id]["mode"] = "chat"
-                user_states[str_chat_id]["history"] =[]
+                user_states[str_chat_id]["history"] = []
+                user_states[str_chat_id]["entry_point"] = "chat"
                 await send_with_keyboard(client, chat_id, "💬 شما وارد بخش **چت با هوش مصنوعی** شدید.\n\nهر سوالی دارید بفرستید تا جواب بدم:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
                 return
 
+            if user_text_str in["/image_analysis", "تحلیل تصویر 🔍"]:
+                user_states[str_chat_id]["mode"] = "chat"
+                user_states[str_chat_id]["history"] = []
+                user_states[str_chat_id]["entry_point"] = "image_analysis"
+                await send_with_keyboard(client, chat_id, "🖼️ شما وارد بخش **تحلیل تصویر اختصاصی** شدید.\n\nلطفاً تصویر خود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
+                return
+
             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)
@@ -3174,12 +3144,6 @@ else:
                 await send_with_keyboard(client, chat_id, clone_text, True)
                 return
 
-            if user_text_str in["/image_analysis", "تحلیل تصویر 🔍"]:
-                user_states[str_chat_id]["mode"] = "image_waiting_for_file"
-                user_states[str_chat_id]["file_bytes"] = None
-                await send_with_keyboard(client, chat_id, "🖼️ شما وارد بخش **تحلیل تصویر اختصاصی** شدید.\n\nلطفاً تصویر خود را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
-                return
-
             if user_text_str in["/stt", "فایل صوتی به متن 📝"]:
                 user_states[str_chat_id]["mode"] = "stt_waiting_for_audio"
                 await send_with_keyboard(client, chat_id, "📝 شما وارد بخش **تبدیل صدا به متن** شدید.\n\nلطفاً فایل خود (ویس، آهنگ، ویدیو و...) را ارسال کنید:\n(برای خروج دکمه «برگشت♻️» را بزنید)", True)
@@ -3356,13 +3320,20 @@ else:
 
             elif current_mode == "chat":
                 if is_file:
-                    await send_with_keyboard(client, chat_id, "📥 در حال دانلود فایل...", False)
+                    dl_msg = "📥 در حال دریافت تصویر..." if user_states[str_chat_id].get("entry_point") == "image_analysis" else "📥 در حال دانلود فایل..."
+                    await send_with_keyboard(client, chat_id, dl_msg, False)
                     try:
                         file_bytes = await helper_download_file(client, msg_obj)
                         user_states[str_chat_id]["file_bytes"] = file_bytes
                         user_states[str_chat_id]["file_name"] = file_name
                         user_states[str_chat_id]["mode"] = "chat_waiting_for_prompt"
-                        await send_with_keyboard(client, chat_id, "✅ فایل با موفقیت دریافت شد.\n\nحالا لطفاً متنی بفرستید و بگویید **چه کاری با این فایل انجام دهم؟**\n(مثلاً: این تصویر را توصیف کن، یا متن این سند را خلاصه کن)", False)
+                        
+                        if user_states[str_chat_id].get("entry_point") == "image_analysis":
+                            msg_text = "✅ تصویر با موفقیت دریافت شد.\n\nحالا لطفاً متنی بگویید **چگونه تحلیل شود؟**\n(می‌توانید سوال خاصی بپرسید یا فقط بخواهید تصویر را توصیف کند)"
+                        else:
+                            msg_text = "✅ فایل با موفقیت دریافت شد.\n\nحالا لطفاً متنی بفرستید و بگویید **چه کاری با این فایل انجام دهم؟**\n(مثلاً: این تصویر را توصیف کن، یا متن این سند را خلاصه کن)"
+                            
+                        await send_with_keyboard(client, chat_id, msg_text, False)
                     except Exception as dl_err: 
                         await send_with_keyboard(client, chat_id, f"❌ خطا در دریافت فایل!\n{str(dl_err)}", False)
                 elif user_text_str: