Update app.py

app.py

@@ -1,10 +1,12 @@
+# --- START OF FILE app.py ---
+
 import os
 import requests
 import json
 from flask import Flask, render_template, request, Response
 import logging
 
-# ================== Logging Setup (Persian) ==================
+# ================== بخش تنظیمات لاگ‌نویسی (فارسی) ==================
 class PersianLogFormatter(logging.Formatter):
     LEVEL_MAP = {
         logging.DEBUG: "دیباگ",
@@ -21,11 +23,14 @@ def setup_logging():
     log_format = '[%(asctime)s] [%(levelname)s]: %(message)s'
     date_format = '%Y-%m-%d %H:%M:%S'
     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)
+    
     root_logger.addHandler(console_handler)
     root_logger.setLevel(logging.INFO)
 
@@ -33,26 +38,27 @@ setup_logging()
 
 app = Flask(__name__)
 
-# URL for your Hugging Face Space API
+# آدرس API اسپیس Hugging Face که به عنوان بک‌اند عمل می‌کند
 HF_API_URL = "https://umint-ai.hf.space/api/chat"
 
 @app.route('/')
 def index():
-    """Serves the main HTML file."""
+    """فایل اصلی HTML را نمایش می‌دهد."""
     return render_template('index.html')
 
 @app.route('/chat', methods=['POST'])
 def chat():
     """
-    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.
+    این اندپوینت به عنوان یک پراکسی امن به اسپیس Hugging Face عمل می‌کند.
+    درخواست چت را از فرانت‌اند دریافت کرده، آن را به اسپیس HF ارسال می‌کند،
+    و پاسخ را بدون تغییر در فرمت SSE به کلاینت بازمی‌گرداند.
     """
     data = request.json
     if not data:
         logging.error("درخواست نامعتبر: JSON وجود ندارد.")
         return Response(json.dumps({"error": "Invalid request"}), status=400, mimetype='application/json')
 
+    # ساختار پیام را مطابق با نیاز اسپیس مقصد آماده می‌کنیم
     payload = {
         "messages": data.get("messages", []),
         "id": data.get("id", "chat_unknown"),
@@ -63,12 +69,13 @@ def chat():
 
     def stream_response():
         try:
+            # ارسال درخواست به اسپیس دیگر و دریافت پاسخ به صورت استریم
             with requests.post(HF_API_URL, json=payload, stream=True, timeout=720) as response:
                 response.raise_for_status()
                 
                 logging.info(f"اتصال موفقیت‌آمیز با HF Space برای چت {payload.get('id')}. در حال استریم پاسخ...")
                 
-                # Forward raw chunks to preserve SSE message boundaries (e.g., '\n\n')
+                # ارسال مستقیم قطعه‌های خام برای حفظ مرزهای پیام SSE (مانند '\n\n')
                 for chunk in response.iter_content(chunk_size=1024):
                     yield chunk
                 
@@ -81,10 +88,13 @@ def chat():
                 "type": "error",
                 "message": "متاسفانه در ارتباط با سرور اصلی خطایی رخ داد. لطفا دوباره تلاش کنید."
             }
+            # ارسال پیام خطا در فرمت SSE
             yield f"data: {json.dumps(error_payload)}\n\n"
 
     return Response(stream_response(), mimetype='text/event-stream')
 
 if __name__ == '__main__':
-    # Use Gunicorn in production instead of this
-    app.run(debug=True, host='0.0.0.0', port=os.environ.get("PORT", 7860))
+    # این بخش فقط برای اجرای محلی است. در هاگینگ فیس از Gunicorn استفاده می‌شود.
+    app.run(debug=True, host='0.0.0.0', port=os.environ.get("PORT", 7860))
+
+# --- END OF FILE app.py ---