Update yml/chat-gemma.py

yml/chat-gemma.py

@@ -4,6 +4,7 @@ from gradio_client import Client
 
 # =========================================================
 # پچ جلوگیری از قطعی ارتباط (Time Out) برای پردازش‌های طولانی
+# زمان انتظار روی ۵ دقیقه تنظیم شده است
 original_client_init = httpx.Client.__init__
 def patched_client_init(self, *args, **kwargs):
     kwargs['timeout'] = httpx.Timeout(300.0)
@@ -25,7 +26,7 @@ gh_run_id = os.environ.get('GITHUB_RUN_ID')
 
 files_to_send = []
 
-# ۱. دانلود فایل و تبدیل مستقیم آن به دتا (Base64 Data URI)
+# ۱. دانلود تصویر و تبدیل قطعی آن به Base64 Data URI (دقیقاً مشابه روش Qwen)
 if file_url and file_url.strip():
     print(f'Downloading attachment for Base64 conversion: {file_url}')
     try:
@@ -36,31 +37,35 @@ if file_url and file_url.strip():
         with open(local_filename, 'wb') as f:
             f.write(req.content)
         
-        # خواندن فایل به صورت باینری و تبدیل به Base64
+        # تبدیل فایل به رشته Base64
         with open(local_filename, "rb") as img_file:
             encoded_string = base64.b64encode(img_file.read()).decode('utf-8')
         
-        # تشخیص نوع فایل برای ساخت هدر دتا
+        # تشخیص حدودی نوع فایل برای هدر
         mime_type = "application/octet-stream"
         if ext.lower() in ['png', 'jpg', 'jpeg', 'webp']:
             mime_type = f"image/{ext.lower()}"
+        elif ext.lower() in ['mp4', 'mov', 'avi', 'webm']:
+            mime_type = f"video/{ext.lower()}"
+        elif ext.lower() in ['wav', 'mp3', 'ogg', 'flac']:
+            mime_type = f"audio/{ext.lower()}"
             
         base64_data_uri = f"data:{mime_type};base64,{encoded_string}"
         
-        # ارسال مستقیم رشته دتا به جای آپلود فیزیکی فایل
+        # ارسال مستقیم رشته متنی بیس۶۴ به عنوان فایل (بدون استفاده از آبجکت دیکشنری)
         files_to_send.append(base64_data_uri)
-        print('   -> Base64 data payload generated successfully.')
+        print('   -> Base64 payload generated successfully.')
     except Exception as e:
-        print(f'Failed to process file to Base64: {e}')
+        print(f'Failed to process file: {e}')
 
 print('Connecting to Gemma-4 Space...')
 bot_reply = ''
 
 try:
-    # ۲. اتصال به اسپیس چت‌بات
+    # ۲. اتصال به اسپیس هدف شما
     client = Client('opera8/gemma-4-e4b-it')
     
-    # ۳. ارسال مستقیم دیکشنری حاوی متن و رشته‌های دتا
+    # ۳. اجرای متد Predict (ارسال پیام به همراه آرایه متنی تصاویر)
     result = client.predict(
         message={'text': prompt, 'files': files_to_send},
         thinking=False,
@@ -70,7 +75,7 @@ try:
         api_name='/generate'
     )
     
-    # ۴. دریافت و استخراج پاسخ نهایی
+    # ۴. استخراج پاسخ متنی
     bot_reply = result[0] if isinstance(result, (list, tuple)) else result
     if isinstance(bot_reply, dict):
         bot_reply = bot_reply.get('text', str(bot_reply))
@@ -83,6 +88,7 @@ except Exception as e:
     else:
         bot_reply = f'⚠️ خطای سرور پردازش: {err_str}'
 
+# بررسی نهایی برای جلوگیری از کرش سیستم در صورت خالی بودن پاسخ
 if not bot_reply or str(bot_reply).strip() == '':
     bot_reply = "⚠️ پاسخی از سمت سرور دریافت نشد."
 else: