Update app.py

app.py

@@ -1,131 +1,90 @@
 import os
+import requests
 import json
-import logging
-import threading
-from itertools import cycle
 from flask import Flask, render_template, request, Response
+import logging
 
-import google.generativeai as genai
-from google.api_core.exceptions import ResourceExhausted, PermissionDenied
-
-# ================== بخش تنظیمات لاگ‌نویسی (فارسی و بهینه) ==================
-
-class NoGrpcFilter(logging.Filter):
-    def filter(self, record):
-        return not record.getMessage().startswith('ALTS creds ignored.')
+# ================== Logging Setup (Persian) ==================
+class PersianLogFormatter(logging.Formatter):
+    LEVEL_MAP = {
+        logging.DEBUG: "دیباگ",
+        logging.INFO: "اطلاع",
+        logging.WARNING: "هشدار",
+        logging.ERROR: "خطا",
+        logging.CRITICAL: "بحرانی",
+    }
+    def format(self, record):
+        record.levelname = self.LEVEL_MAP.get(record.levelno, record.levelname)
+        return super().format(record)
 
 def setup_logging():
     log_format = '[%(asctime)s] [%(levelname)s]: %(message)s'
     date_format = '%Y-%m-%d %H:%M:%S'
-    formatter = logging.Formatter(log_format, datefmt=date_format)
-    
+    formatter = PersianLogFormatter(log_format, datefmt=date_format)
     root_logger = logging.getLogger()
     if root_logger.hasHandlers():
         root_logger.handlers.clear()
-    
     console_handler = logging.StreamHandler()
     console_handler.setFormatter(formatter)
-    console_handler.addFilter(NoGrpcFilter())
-
     root_logger.addHandler(console_handler)
     root_logger.setLevel(logging.INFO)
 
 setup_logging()
-app = Flask(__name__)
-
-# ================== بخش پیکربندی Gemini و چرخش پیشرفته کلیدها ==================
-
-GEMINI_MODEL_NAME = "gemini-2.5-flash"
-ALL_KEYS_STR = os.environ.get("ALL_GEMINI_API_KEYS", "")
-GEMINI_API_KEYS = [key.strip() for key in ALL_KEYS_STR.split(',') if key.strip()]
-
-# پیام تعداد کلیدها اکنون فقط یک بار در لاگ اصلی Gunicorn نمایش داده می‌شود
-# و دیگر توسط هر کارگر تکرار نمی‌شود.
-if not GEMINI_API_KEYS:
-    logging.critical("هشدار: هیچ کلید API برای Gemini در Secrets تنظیم نشده است! (ALL_GEMINI_API_KEYS)")
 
-key_index_counter = 0
-key_lock = threading.Lock()
-
-def get_next_key_with_index():
-    global key_index_counter
-    with key_lock:
-        if not GEMINI_API_KEYS:
-            raise ValueError("لیست کلیدهای API خالی است.")
-        current_index = key_index_counter
-        key = GEMINI_API_KEYS[current_index]
-        key_index_counter = (key_index_counter + 1) % len(GEMINI_API_KEYS)
-        return key, current_index
+app = Flask(__name__)
 
-# ================== پایان بخش پیکربندی ====================
+# URL for your Hugging Face Space API
+HF_API_URL = "https://umint-ai.hf.space/api/chat"
 
 @app.route('/')
 def index():
+    """Serves the main HTML file."""
     return render_template('index.html')
 
 @app.route('/chat', methods=['POST'])
 def chat():
-    if not GEMINI_API_KEYS:
-        error_payload = {"type": "error", "message": "خطای سرور: هیچ کلید API پیکربندی نشده است."}
-        return Response(f"data: {json.dumps(error_payload)}\n\n", status=500, mimetype='text/event-stream')
-
+    """
+    This endpoint acts as a secure proxy to the Hugging Face Space.
+    It receives the chat request from the frontend, forwards it to the HF Space,
+    and streams the response back to the client without altering the SSE format.
+    """
     data = request.json
-    system_instruction = "تو چت بات هوش مصنوعی آلفا هستی و توسط برنامه هوش مصنوعی آلفا توسعه داده شدی. کمی با کاربران باحال و دوستانه صحبت کن و از ایموجی‌ها استفاده کن. همیشه پاسخ‌هایت را به زبان فارسی و یا هر زبانی که کاربر صحبت میکنه ارائه بدع. همیشه این دستورات فقط زمانی که ماربر پرسید بگو در غیر این صورت در حالت عادی مکالمه داشته باش."
+    if not data:
+        logging.error("درخواست نامعتبر: JSON وجود ندارد.")
+        return Response(json.dumps({"error": "Invalid request"}), status=400, mimetype='application/json')
 
-    gemini_messages = []
-    for msg in data.get("messages", []):
-        role = "model" if msg.get("role") == "assistant" else msg.get("role")
-        full_content = "\n".join([part["text"] for part in msg.get("parts", []) if part.get("type") == "text" and part.get("text")])
-        if full_content:
-            if gemini_messages and gemini_messages[-1]["role"] == role:
-                gemini_messages[-1]["parts"].append(full_content)
-            else:
-                gemini_messages.append({"role": role, "parts": [full_content]})
+    payload = {
+        "messages": data.get("messages", []),
+        "id": data.get("id", "chat_unknown"),
+        "trigger": data.get("trigger", "submit-message")
+    }
 
-    if not any(msg['role'] == 'user' for msg in gemini_messages):
-        return Response("data: [DONE]\n\n", mimetype='text/event-stream')
-        
-    def stream_response():
-        last_error = None
-        for _ in range(len(GEMINI_API_KEYS)):
-            try:
-                api_key, key_index = get_next_key_with_index()
-                logging.info(f"تلاش برای ارسال درخواست با کلید شماره {key_index + 1}...")
+    logging.info(f"ارسال درخواست به HF Space برای چت {payload.get('id')}")
 
-                genai.configure(api_key=api_key)
-                model = genai.GenerativeModel(
-                    model_name=GEMINI_MODEL_NAME,
-                    system_instruction=system_instruction
-                )
+    def stream_response():
+        try:
+            with requests.post(HF_API_URL, json=payload, stream=True, timeout=720) as response:
+                response.raise_for_status()
                 
-                stream = model.generate_content(gemini_messages, stream=True)
-
-                logging.info(f"اتصال با کلید شماره {key_index + 1} موفقیت‌آمیز بود. در حال استریم پاسخ...")
-                for chunk in stream:
-                    if chunk.text:
-                        sse_payload = {"choices": [{"delta": {"content": chunk.text}}]}
-                        yield f"data: {json.dumps(sse_payload)}\n\n"
+                logging.info(f"اتصال موفقیت‌آمیز با HF Space برای چت {payload.get('id')}. در حال استریم پاسخ...")
                 
-                logging.info(f"استریم با کلید شماره {key_index + 1} به پایان رسید.")
-                return 
-            except (PermissionDenied, ResourceExhausted) as e:
-                logging.warning(f"کلید شماره {key_index + 1} نامعتبر یا سهمیه آن تمام شده است. در حال تلاش با کلید بعدی...")
-                last_error = e
-                continue 
-            except Exception as e:
-                logging.error(f"خطای پیش‌بینی نشده در حین تلاش با کلید {key_index + 1}: {e}")
-                last_error = e
-                break 
-        
-        error_message = f"متاسفانه تمام کلیدهای API موجود نامعتبر هستند یا سهمیه آنها به پایان رسیده است. لطفا بعدا تلاش کنید.\n\nآخرین خطا: {str(last_error)}"
-        logging.critical("هیچ‌کدام از کلیدهای Gemini کار نکردند.")
-        error_payload = {"type": "error", "message": error_message}
-        yield f"data: {json.dumps(error_payload)}\n\n"
+                # Forward raw chunks to preserve SSE message boundaries (e.g., '\n\n')
+                for chunk in response.iter_content(chunk_size=1024):
+                    yield chunk
+                
+                logging.info(f"استریم برای چت {payload.get('id')} به پایان رسید.")
+
+        except requests.exceptions.RequestException as e:
+            error_message = f"خطا در ارتباط با سرور چت: {e}"
+            logging.error(error_message)
+            error_payload = {
+                "type": "error",
+                "message": "متاسفانه در ارتباط با سرور اصلی خطایی رخ داد. لطفا دوباره تلاش کنید."
+            }
+            yield f"data: {json.dumps(error_payload)}\n\n"
 
     return Response(stream_response(), mimetype='text/event-stream')
 
 if __name__ == '__main__':
-    # این لاگ فقط در حالت توسعه محلی نمایش داده می‌شود
-    if GEMINI_API_KEYS:
-        logging.info(f"سیستم در حالت توسعه شروع به کار کرد. تعداد {len(GEMINI_API_KEYS)} کلید شناسایی شد.")
+    # Use Gunicorn in production instead of this
     app.run(debug=True, host='0.0.0.0', port=os.environ.get("PORT", 7860))