Update main.py

main.py

@@ -1186,6 +1186,31 @@ async def process_legacy_vc_job(client, chat_id, src_bytes, model_url, pitch, mo
 # ==============================================================================
 # 🟢 پارت 13: پردازش هوش مصنوعی متنی و تصویری (ترکیب کامل با Aya Expanse و Gemma)
 # ==============================================================================
+import re
+
+def clean_math_formatting(text):
+    """پاکسازی و خوانا کردن فرمول‌های ریاضی و علائم لاتک (LaTeX) برای پیام‌رسان روبیکا"""
+    if not text: return text
+    
+    # فاصله دادن بین عدد صحیح و کسر (مثلاً 3\frac تبدیل شود به 3 \frac)
+    text = re.sub(r'(\d)\\frac', r'\1 \\frac', text)
+    
+    # تبدیل کسرها به شکل خطی ساده (a/b)
+    for _ in range(3):
+        text = re.sub(r'\\frac\{([^{}]*)\}\{([^{}]*)\}', r'\1/\2', text)
+    
+    replacements = {
+        '\\times': '×', '\\div': '÷', '\\pm': '±', '\\cdot': '⋅',
+        '\\left(': '(', '\\right)': ')', '\\left[': '[', '\\right]': ']',
+        '\\left{': '{', '\\right}': '}', '\\sqrt': '√', '\\pi': 'π', 
+        '\\approx': '≈', '\\neq': '≠', '\\leq': '≤', '\\geq': '≥', 
+        '\\infty': '∞', '$$': '', '$': '', '\\[': '', '\\]': '', '\\(': '', '\\)': ''
+    }
+    
+    for old, new in replacements.items():
+        text = text.replace(old, new)
+        
+    return text.strip()
 
 async def helper_upload_to_aya(file_bytes, file_name, mime_type):
     """آپلود تصویر در سرور بکاپ (Aya) در صورت قطعی گیت‌هاب"""
@@ -1515,6 +1540,9 @@ async def process_gemini(client, chat_id, prompt, file_bytes=None, file_name=Non
         await send_with_keyboard(client, chat_id, "❌ تمامی سرورها شلوغ هستند یا حجم فایل بالا بود. لطفاً بعداً امتحان کنید.", False)
         return
 
+    # 🧹 اعمال فیلتر پاکسازی فرمول‌های ریاضی و لاتک روی پاسخ نهایی
+    final_answer = clean_math_formatting(final_answer)
+
     # ذخیره پاسخ در تاریخچه چت کاربر
     history.append({"role": "model", "parts": [{"text": final_answer}]})
     user_states[chat_id]["history"] = history
@@ -2009,6 +2037,8 @@ async def process_stt(client, chat_id, audio_bytes, file_name):
     except Exception: pass
 
     if transcribed_text:
+        # 🧹 اعمال فیلتر پاکسازی فرمول‌های ریاضی و لاتک
+        transcribed_text = clean_math_formatting(transcribed_text)
         sent = await send_with_keyboard(client, chat_id, f"📝 **متن استخراج شده:**\n\n{transcribed_text}", True)
         if sent and not creds.get("is_premium"):
             user_credits_db[str_chat_id]["stt"] -= 1
@@ -2045,6 +2075,8 @@ async def process_image_analysis(client, chat_id, file_bytes, file_name, prompt)
     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