Update yml/chat-gemma.py

yml/chat-gemma.py

@@ -1,117 +1,38 @@
-import os, sys, requests, time, base64
-import httpx
-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)
-    original_client_init(self, *args, **kwargs)
-httpx.Client.__init__ = patched_client_init
-
-original_async_init = httpx.AsyncClient.__init__
-def patched_async_init(self, *args, **kwargs):
-    kwargs['timeout'] = httpx.Timeout(300.0)
-    original_async_init(self, *args, **kwargs)
-httpx.AsyncClient.__init__ = patched_async_init
-# =========================================================
+import os, sys, requests, base64
 
 prompt = os.environ.get('PROMPT', '')
 file_url = os.environ.get('FILE_URL', '')
 run_id = os.environ.get('RUN_ID')
-space_url = os.environ.get('SPACE_URL')
-gh_run_id = os.environ.get('GITHUB_RUN_ID')
-
-files_to_send = []
+space_url = "https://opera8-gemma-4-e4b-it.hf.space" # آدرس مستقیم اسپیس شما
 
-# ۱. دانلود تصویر و تبدیل قطعی آن به Base64 Data URI (دقیقاً مشابه روش Qwen)
-if file_url and file_url.strip():
-    print(f'Downloading attachment for Base64 conversion: {file_url}')
-    try:
-        req = requests.get(file_url, timeout=30)
-        ext = file_url.split('.')[-1] if '.' in file_url else 'bin'
-        local_filename = f'attachment.{ext}'
-        
-        with open(local_filename, 'wb') as f:
-            f.write(req.content)
-        
-        # تبدیل فایل به رشته 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 payload generated successfully.')
-    except Exception as e:
-        print(f'Failed to process file: {e}')
+image_data = None
+if file_url:
+    # دانلود و تبدیل به base64
+    img_resp = requests.get(file_url)
+    image_data = base64.b64encode(img_resp.content).decode('utf-8')
 
-print('Connecting to Gemma-4 Space...')
-bot_reply = ''
+payload = {
+    "prompt": prompt,
+    "image_base64": image_data,
+    "system_prompt": "شما یک دستیار هوشمند هستید."
+}
 
 try:
-    # ۲. اتصال به اسپیس هدف شما
-    client = Client('opera8/gemma-4-e4b-it')
+    # ارسال به FastAPI
+    response = requests.post(f"{space_url}/chat", json=payload, timeout=120)
+    result = response.json()
     
-    # ۳. اجرای متد Predict (ارسال پیام به همراه آرایه متنی تصاویر)
-    result = client.predict(
-        message={'text': prompt, 'files': files_to_send},
-        thinking=False,
-        max_new_tokens=2000,
-        max_soft_tokens=280,
-        system_prompt='شما یک دستیار هوش مصنوعی بسیار باهوش و مودب به زبان فارسی هستید.',
-        api_name='/generate'
-    )
+    bot_reply = result.get("response", "⚠️ خطایی رخ داد.")
     
-    # ۴. استخراج پاسخ متنی
-    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))
-        
-except Exception as e:
-    err_str = str(e)
-    print(f'Generation Error: {err_str}')
-    if 'quota' in err_str.lower() or '429' in err_str:
-        bot_reply = '⚠️ ظرفیت پردازش موقتاً پر شده است. لطفاً کمی بعد امتحان کنید.'
-    else:
-        bot_reply = f'⚠️ خطای سرور پردازش: {err_str}'
-
-# بررسی نهایی برای جلوگیری از کرش سیستم در صورت خالی بودن پاسخ
-if not bot_reply or str(bot_reply).strip() == '':
-    bot_reply = "⚠️ پاسخی از سمت سرور دریافت نشد."
-else:
-    bot_reply = str(bot_reply)
-
-print('Uploading response back to Docker Space...')
-try:
+    # آپلود پاسخ به سرور خودتان (Docker Space)
     with open('response.txt', 'w', encoding='utf-8') as f:
         f.write(bot_reply)
-        
-    with open('response.txt', 'rb') as f:
-        res = requests.post(
-            f'{space_url}/api/webhook/upload',
-            data={'run_id': run_id, 'github_run_id': gh_run_id, 'ext': 'txt'},
-            files={'file': f}
-        )
     
-    if res.status_code == 200:
-        print('SUCCESS!')
-    else:
-        print(f"Upload failed with status {res.status_code}: {res.text}")
-        sys.exit(1)
-        
+    requests.post(
+        f"{os.environ.get('SPACE_URL')}/api/webhook/upload",
+        data={'run_id': run_id, 'github_run_id': os.environ.get('GITHUB_RUN_ID'), 'ext': 'txt'},
+        files={'file': open('response.txt', 'rb')}
+    )
 except Exception as e:
-    print(f"Upload Exception: {e}")
+    print(f"Error: {e}")
     sys.exit(1)