Add files via upload

README.md

@@ -0,0 +1,10 @@
+---
+title: Chatlala
+emoji: 🏃
+colorFrom: red
+colorTo: green
+sdk: docker
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,223 @@
+# --- START OF FILE app.py ---
+
+import os
+import json
+import logging
+import threading
+import base64
+import io
+from flask import Flask, render_template, request, Response
+import requests
+import docx
+
+# ================== بخش تنظیمات لاگ‌نویسی (بدون تغییر) ==================
+
+class NoGrpcFilter(logging.Filter):
+    def filter(self, record):
+        return not record.getMessage().startswith('ALTS creds ignored.')
+
+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)
+    
+    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()]
+
+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
+
+# *** START: MODIFIED - افزودن ثابت‌های جدید برای مهلت زمانی ***
+# مهلت زمانی (به ثانیه) برای شروع دریافت پاسخ از سرور
+STREAM_START_TIMEOUT = 4
+# مهلت زمانی (به ثانیه) برای دریافت هر قطعه جدید از داده در حین استریم
+# اگر در این مدت داده جدیدی نرسد، اتصال قطع و با کلید بعدی تلاش می‌شود
+STREAM_READ_TIMEOUT = 15
+# *** END: MODIFIED ***
+
+# ================== پایان بخش پیکربندی ====================
+
+@app.route('/')
+def index():
+    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')
+
+    data = request.json
+    system_instruction = "تو چت بات هوش مصنوعی آلفا هستی و توسط برنامه هوش مصنوعی آلفا توسعه داده شدی. کمی با کاربران باحال و دوستانه صحبت کن و از ایموجی‌ها استفاده کن. همیشه پاسخ‌هایت را به زبان فارسی و یا هر زبانی که کاربر صحبت میکنه ارائه بده."
+    
+    show_thoughts = data.get("show_thoughts", False)
+
+    # بخش پردازش پیام‌ها و فایل DOCX (بدون تغییر باقی می‌ماند)
+    gemini_messages = []
+    for msg in data.get("messages", []):
+        role = "model" if msg.get("role") == "assistant" else msg.get("role")
+        
+        processed_parts = []
+        for part in msg.get("parts", []):
+            if part.get("text"):
+                processed_parts.append({"text": part["text"]})
+            
+            if part.get("base64Data") and part.get("mimeType"):
+                mime_type = part["mimeType"]
+                
+                if mime_type == "application/vnd.openxmlformats-officedocument.wordprocessingml.document":
+                    try:
+                        decoded_data = base64.b64decode(part["base64Data"])
+                        file_stream = io.BytesIO(decoded_data)
+                        document = docx.Document(file_stream)
+                        full_text = "\n".join([para.text for para in document.paragraphs])
+                        
+                        final_text_part = f"کاربر یک فایل Word آپلود کرد. محتوای متنی آن به شرح زیر است:\n\n---\n\n{full_text}\n\n---"
+                        processed_parts.append({"text": final_text_part})
+                        logging.info("فایل DOCX با موفقیت پردازش و متن آن استخراج شد.")
+
+                    except Exception as e:
+                        logging.error(f"خطا در پردازش فایل DOCX: {e}")
+                        processed_parts.append({"text": "[خطا: امکان پردازش فایل Word وجو�� نداشت.]"})
+                
+                else:
+                    processed_parts.append({"inline_data": {"mime_type": part["mimeType"], "data": part["base64Data"]}})
+        
+        if processed_parts:
+            if gemini_messages and gemini_messages[-1]["role"] == role:
+                gemini_messages[-1]["parts"].extend(processed_parts)
+            else:
+                gemini_messages.append({"role": role, "parts": processed_parts})
+
+    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}...")
+
+                api_endpoint = f"https://generativelanguage.googleapis.com/v1beta/models/{GEMINI_MODEL_NAME}:streamGenerateContent?key={api_key}&alt=sse"
+                
+                payload = {
+                    "contents": gemini_messages,
+                    "systemInstruction": {"parts": [{"text": system_instruction}]},
+                    "tools": [{"google_search": {}}],
+                    "generationConfig": {
+                        "temperature": 0.7,
+                    }
+                }
+                
+                if show_thoughts:
+                    payload["generationConfig"]["thinking_config"] = {
+                        "include_thoughts": True
+                    }
+
+                # *** START: MODIFIED - استفاده از Timeout تفکیک شده ***
+                # timeout اول برای اتصال اولیه، timeout دوم برای فاصله بین دریافت داده‌ها
+                with requests.post(api_endpoint, json=payload, stream=True, timeout=(STREAM_START_TIMEOUT, STREAM_READ_TIMEOUT)) as response:
+                # *** END: MODIFIED ***
+                    if response.status_code == 429:
+                        logging.warning(f"کلید شماره {key_index + 1} سهمیه آن تمام شده است. در حال تلاش با کلید بعدی...")
+                        last_error = "Rate limit exceeded"
+                        continue
+                    
+                    response.raise_for_status()
+                    logging.info(f"اتصال با کلید شماره {key_index + 1} موفقیت‌آمیز بود. در حال استریم پاسخ...")
+                    
+                    # بخش استریم پاسخ (بدون تغییر)
+                    for line in response.iter_lines():
+                        if line:
+                            decoded_line = line.decode('utf-8')
+                            if decoded_line.startswith('data: '):
+                                try:
+                                    chunk_data = json.loads(decoded_line[6:])
+                                    parts = chunk_data.get("candidates", [{}])[0].get("content", {}).get("parts", [])
+                                    
+                                    for part in parts:
+                                        if "text" not in part or not part["text"]:
+                                            continue
+                                        is_a_thought = part.get("thought") is True
+                                        if show_thoughts and is_a_thought:
+                                            thought_payload = {"type": "thought", "content": part["text"]}
+                                            yield f"data: {json.dumps(thought_payload)}\n\n"
+                                        elif not is_a_thought:
+                                            sse_payload = {"choices": [{"delta": {"content": part["text"]}}]}
+                                            yield f"data: {json.dumps(sse_payload)}\n\n"
+                                except (json.JSONDecodeError, IndexError, KeyError):
+                                    continue
+                    
+                    logging.info(f"استریم با کلید شماره {key_index + 1} به پایان رسید.")
+                    return
+
+            # *** START: MODIFIED - افزودن ReadTimeout به مدیریت خطا ***
+            except (requests.exceptions.Timeout, requests.exceptions.ReadTimeout) as e:
+                if isinstance(e, requests.exceptions.ReadTimeout):
+                    logging.warning(f"استریم با کلید شماره {key_index + 1} به دلیل عدم دریافت داده جدید متوقف شد (ReadTimeout). در حال تلاش با کلید بعدی...")
+                    last_error = f"ReadTimeout after {STREAM_READ_TIMEOUT} seconds of inactivity"
+                else:
+                    logging.warning(f"مهلت زمانی برای اتصال با کلید شماره {key_index + 1} به پایان رسید (ConnectTimeout). در حال تلاش با کلید بعدی...")
+                    last_error = f"ConnectTimeout after {STREAM_START_TIMEOUT} seconds"
+                continue
+            # *** END: MODIFIED ***
+
+            except requests.exceptions.HTTPError as e:
+                if e.response.status_code == 403:
+                     logging.warning(f"کلید شماره {key_index + 1} نامعتبر است (Permission Denied). در حال تلاش با کلید بعدی...")
+                     last_error = e
+                     continue
+                else:
+                     logging.error(f"خطای HTTP پیش‌بینی نشده در حین تلاش با کلید {key_index + 1}: {e}")
+                     last_error = e
+                     break
+
+            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"
+
+    return Response(stream_response(), mimetype='text/event-stream')
+
+if __name__ == '__main__':
+    if GEMINI_API_KEYS:
+        logging.info(f"سیستم در حالت توسعه شروع به کار کرد. تعداد {len(GEMINI_API_KEYS)} کلید شناسایی شد.")
+    app.run(debug=True, host='0.0.0.0', port=os.environ.get("PORT", 7860))
+
+# --- END OF FILE app.py ---

requirements.txt

@@ -0,0 +1,13 @@
+# --- START OF FILE requirements.txt ---
+
+Flask==2.2.2
+Werkzeug==2.2.2
+gunicorn==20.1.0
+requests==2.28.1
+gevent==22.10.2
+sseclient-py==1.8.0
+filelock==3.13.1
+google-generativeai
+python-docx==1.1.2 # *** کتابخانه جدید برای خواندن فایل‌های Word اضافه شد ***
+
+# --- END OF FILE requirements.txt ---