Create app.py

app.py

@@ -0,0 +1,271 @@
+import os
+import requests
+from flask import Flask, Response, request, stream_with_context
+from urllib.parse import urlparse, urljoin, urlunparse
+import re
+import random
+import logging
+import json # برای تبدیل دیکشنری پایتون به رشته JSON
+
+app = Flask(__name__) # اصلاح: __name__ به جای name
+SECRET_NAME_FOR_TARGET_URL = "TARGET_HF_SPACE_URL"
+TARGET_URL_FROM_SECRET = os.environ.get(SECRET_NAME_FOR_TARGET_URL)
+REQUEST_TIMEOUT = 45
+
+# --- Logging Setup (حداقل لاگ ممکن) ---
+log = logging.getLogger('werkzeug')
+log.setLevel(logging.ERROR) # فقط خطاهای Werkzeug را لاگ کن
+# app.logger.setLevel(logging.CRITICAL + 1) # غیرفعال کردن کامل لاگ‌های اپلیکیشن Flask
+# برای دیباگ می‌توانید این خط را کامنت کنید یا سطح را تغییر دهید
+if not (os.environ.get('FLASK_DEBUG', '0') == '1' or os.environ.get('FLASK_ENV') == 'development'):
+    app.logger.setLevel(logging.CRITICAL + 1)
+else:
+    app.logger.setLevel(logging.INFO) # لاگ در حالت دیباگ
+
+print(f"APP_STARTUP: TARGET_URL_FROM_SECRET: {'SET' if TARGET_URL_FROM_SECRET else 'NOT SET'}")
+if TARGET_URL_FROM_SECRET:
+    print(f"APP_STARTUP: Target URL is '{TARGET_URL_FROM_SECRET[:20]}...'") # بخشی از URL برای تایید
+
+COMMON_USER_AGENTS = [
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/102.0.5005.63 Safari/537.36",
+    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/15.5 Safari/605.1.15",
+    # ... (سایر User-Agent ها)
+]
+
+SIMPLE_TEXT_REPLACEMENTS = {
+    "Aya Vision": "تحلیل",
+    "Chat with Aya": "چت🤖",
+    "Visualize with Aya": "عکس",
+    "Speak with Aya": "", # این مورد باعث مخفی شدن المان والدش می‌شود
+    # ... (سایر جایگزینی‌های متن)
+    "Gradio": "App", # مثال: جایگزینی کلمه Gradio
+    "Built with Gradio": "" # حذف این عبارت
+}
+
+GRADIO_HIDE_CSS_STANDARD = """
+<style>
+  /* فوتر Gradio */
+  .gradio-container .meta-footer, .gradio-container footer, div[class*="footer"], footer, a[href*="gradio.app"],
+  /* دکمه Settings Gradio */
+  .gradio-container button[id*="settings"], .gradio-container div[class*="settings-button"],
+  a[href*="gradio.app/"], button[title*="Settings"], button[aria-label*="Settings"], div[data-testid*="settings"] {
+    display: none !important;
+    visibility: hidden !important;
+    opacity: 0 !important;
+    width: 0 !important;
+    height: 0 !important;
+    overflow: hidden !important;
+    margin: 0 !important;
+    padding: 0 !important;
+    border: none !important;
+    font-size: 0 !important;
+    line-height: 0 !important;
+  }
+  /* کلاس برای مخفی کردن المان‌هایی که متن آنها با رشته خالی جایگزین شده */
+  .gr-proxy-item-hidden-by-text {
+    display: none !important;
+    visibility: hidden !important;
+  }
+</style>
+"""
+
+def process_html_content_server_side(html_string):
+    """فقط حذف‌های اولیه سمت سرور با Regex (بدون تزریق CSS در اینجا)."""
+    processed_html = html_string
+    # ... (کد مربوط به حذف با Regex شما) ...
+    return processed_html
+
+@app.route('/', defaults={'path': ''}, methods=['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'OPTIONS', 'HEAD'])
+@app.route('/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'OPTIONS', 'HEAD']) # اصلاح: <path:path>
+def proxy_request_handler(path):
+    if not TARGET_URL_FROM_SECRET:
+        app.logger.critical(f"CRITICAL: Secret '{SECRET_NAME_FOR_TARGET_URL}' is not set.")
+        return "Proxy Configuration Error: Target URL secret is not configured.", 500
+
+    base_target_url = TARGET_URL_FROM_SECRET.rstrip('/')
+    # اطمینان از اینکه path با / شروع نمی‌شود اگر base_target_url خودش با / ختم شده
+    target_full_url = urljoin(base_target_url + "/", path.lstrip('/'))
+    if request.query_string:
+        target_full_url += "?" + request.query_string.decode()
+    
+    app.logger.debug(f"Proxying request for path: /{path} to {target_full_url}")
+
+    try:
+        parsed_target_url_for_host = urlparse(TARGET_URL_FROM_SECRET)
+        target_hostname = parsed_target_url_for_host.hostname
+        user_agent_to_send = random.choice(COMMON_USER_AGENTS)
+        
+        excluded_incoming_headers = [
+            'host', 'cookie', 'connection', 'upgrade-insecure-requests', 
+            'if-none-match', 'if-modified-since', 'referer', 'x-hf-space-host',
+            'content-length' # اجازه به requests برای محاسبه مجدد
+        ]
+        forward_headers = {key: value for key, value in request.headers.items() if key.lower() not in excluded_incoming_headers}
+        forward_headers['User-Agent'] = user_agent_to_send
+        forward_headers['Host'] = target_hostname
+        forward_headers['Accept-Encoding'] = 'gzip, deflate, br'
+        
+        # مدیریت X-Forwarded-For
+        if request.headers.getlist("X-Forwarded-For"):
+            forward_headers["X-Forwarded-For"] = request.headers.getlist("X-Forwarded-For")[0] + ", " + request.remote_addr
+        else:
+            forward_headers["X-Forwarded-For"] = request.remote_addr
+        
+        # برای POST, PUT, etc.
+        request_body = request.get_data() if request.method not in ['GET', 'HEAD', 'OPTIONS'] else None
+
+        with requests.Session() as s:
+            target_response = s.request(
+                method=request.method,
+                url=target_full_url,
+                headers=forward_headers,
+                data=request_body,
+                stream=True, # مهم برای stream کردن پاسخ
+                timeout=REQUEST_TIMEOUT,
+                allow_redirects=False # ریدایرکت‌ها را خودمان مدیریت می‌کنیم
+            )
+
+        # مدیریت ریدایرکت‌ها
+        if 300 <= target_response.status_code < 400 and 'Location' in target_response.headers:
+            location = target_response.headers['Location']
+            # ... (کد مدیریت ریدایرکت شما، اطمینان از بازنویسی صحیح Location) ...
+            # این بخش نیاز به بازبینی دقیق دارد تا Location به درستی به دامنه پروکسی بازنویسی شود
+            parsed_location = urlparse(location)
+            rewritten_location_header_val = location
+            # اگر ریدایرکت نسبی است یا به همان هاست هدف است، آن را برای پروکسی بازنویسی کن
+            if not parsed_location.scheme or parsed_location.hostname == target_hostname:
+                # URL کامل ریدایرکت را بر اساس URL نهایی که requests به آن رسیده، بساز
+                abs_target_redirect_url = urljoin(target_response.url, location)
+                new_path_on_target = urlparse(abs_target_redirect_url).path
+                new_query_on_target = urlparse(abs_target_redirect_url).query
+                
+                rewritten_location_path = new_path_on_target
+                if new_query_on_target:
+                    rewritten_location_path += "?" + new_query_on_target
+                # اطمینان از اینکه با / شروع می‌شود اگر خالی نیست
+                if not rewritten_location_path.startswith('/') and rewritten_location_path:
+                    rewritten_location_path = '/' + rewritten_location_path
+                rewritten_location_header_val = rewritten_location_path if rewritten_location_path else "/"
+
+            final_redirect_headers = {'Location': rewritten_location_header_val}
+            # سایر هدرهای مهم ریدایرکت را هم منتقل کن
+            for h_key, h_val in target_response.headers.items():
+                if h_key.lower() in ['set-cookie', 'cache-control', 'expires', 'pragma', 'vary']:
+                    final_redirect_headers[h_key] = h_val
+            return Response(response=None, status=target_response.status_code, headers=final_redirect_headers)
+
+        # مدیریت خطاهای سرور هدف
+        if target_response.status_code >= 400:
+            app.logger.warning(f"Target {target_full_url} error: {target_response.status_code}")
+            error_content = target_response.content # ممکن است بزرگ باشد
+            error_headers_from_target = {
+                k:v for k,v in target_response.headers.items() 
+                if k.lower() not in ['content-encoding', 'transfer-encoding', 'content-length']
+            }
+            return Response(response=error_content, status=target_response.status_code, headers=error_headers_from_target)
+
+        content_type = target_response.headers.get('Content-Type', 'application/octet-stream')
+        
+        excluded_response_headers = [
+            'content-encoding', 'transfer-encoding', 'connection', 'keep-alive',
+            'x-frame-options', 'strict-transport-security', 'public-key-pins',
+            'content-length', 'server', 'x-powered-by', 'date',
+            'link',  # <<< مهم: حذف هدر Link از پاسخ
+            'x-canonical-url' # <<< مهم: حذف هدر احتمالی دیگر برای URL کنونیکال
+        ]
+        final_response_headers = {
+            key: value for key, value in target_response.headers.items() 
+            if key.lower() not in excluded_response_headers
+        }
+        final_response_headers['Content-Type'] = content_type
+        final_response_headers['Cache-Control'] = 'no-store, no-cache, must-revalidate, max-age=0'
+        final_response_headers['Pragma'] = 'no-cache'
+        final_response_headers['Expires'] = '0'
+        final_response_headers['Referrer-Policy'] = 'strict-origin-when-cross-origin'
+
+        def generate_response_content_stream():
+            # ... (کد طولانی شما برای generate_response_content_stream با تزریق CSS/JS) ...
+            # این بخش بسیار طولانی است و برای سادگی اینجا خلاصه شده.
+            # اگر این کد Flask اجرا شود، این تابع مسئول تغییر محتوای HTML خواهد بود.
+            # در حال حاضر با تنظیمات Nginx، این تابع اجرا نمی‌شود.
+            is_html_content_type = 'text/html' in content_type.lower()
+            base_tag_injected_flag = False
+            css_and_js_injected_flag = False
+            
+            current_buffer = b''
+            charset_match = re.search(r'charset=([\w-]+)', content_type, re.IGNORECASE)
+            html_processing_encoding = charset_match.group(1) if charset_match else 'utf-8'
+
+            # محاسبه base href (همانطور که داشتید)
+            # ...
+            # base_tag_html_to_inject = f'<base href="{escaped_href_for_final_base_tag}">'
+
+            # js_replacements_json = json.dumps(SIMPLE_TEXT_REPLACEMENTS)
+            # mutation_observer_script = f"""...""" # اسکریپت JS شما
+
+            try:
+                for chunk_data in target_response.iter_content(chunk_size=8192): # chunk_size می‌تواند تنظیم شود
+                    if not chunk_data: continue
+
+                    if is_html_content_type:
+                        current_buffer += chunk_data
+                        # ... (منطق پردازش بافر و تزریق HTML/CSS/JS شما) ...
+                        # مثال ساده:
+                        # buffer_as_string = current_buffer.decode(html_processing_encoding, errors='replace')
+                        # processed_chunk_string = buffer_as_string 
+                        # if not css_and_js_injected_flag and '<head>' in processed_chunk_string:
+                        #    # تزریق CSS و base tag
+                        #    css_and_js_injected_flag = True
+                        # if '</body>' in processed_chunk_string:
+                        #    # تزریق JS
+                        # yield processed_chunk_string.encode(html_processing_encoding, errors='replace')
+                        # current_buffer = b''
+                        # این بخش باید کامل از کد شما کپی شود اگر می‌خواهید از Flask استفاده کنید.
+                        # برای سادگی، اینجا فقط محتوای خام را برمی‌گردانیم:
+                        yield chunk_data # در حالت فعلی، برای اینکه کد خلاصه شود.
+                                         # اگر این کد Flask فعال شود، باید منطق پردازش HTML شما اینجا باشد.
+                    else: # محتوای غیر HTML
+                        yield chunk_data
+                
+                # پردازش باقیمانده بافر (اگر چیزی مانده)
+                if current_buffer: 
+                    if is_html_content_type:
+                        # ... (منطق پردازش باقیمانده بافر HTML شما) ...
+                        yield current_buffer # باز هم، برای خلاصه بودن.
+                    else:
+                        yield current_buffer
+            except Exception as stream_err:
+                app.logger.error(f"Error during HTML/Stream processing for {target_full_url}: {stream_err}")
+            finally:
+                target_response.close()
+
+        if request.method == 'HEAD':
+            return Response(response=None, status=target_response.status_code, headers=final_response_headers)
+        else:
+            return Response(stream_with_context(generate_response_content_stream()), status=target_response.status_code, headers=final_response_headers)
+
+    except requests.exceptions.Timeout:
+        app.logger.error(f"Timeout ({REQUEST_TIMEOUT}s) fetching {target_full_url}")
+        return "Error: Request to target site timed out.", 504
+    except requests.exceptions.HTTPError as http_err:
+        app.logger.error(f"HTTPError {http_err.response.status_code} from {target_full_url}. Resp: {http_err.response.text[:200]}")
+        # ... (کد مدیریت خطا) ...
+        return "HTTP Error from target.", 502 # یا وضعیت خطای اصلی
+    except requests.exceptions.ConnectionError as conn_err:
+        app.logger.error(f"ConnectionError for {target_full_url}: {conn_err}")
+        return f"Error: Could not connect to target.", 502
+    except requests.exceptions.RequestException as req_err:
+        app.logger.error(f"RequestException for {target_full_url}: {req_err}")
+        return f"Error fetching content.", 502
+    except Exception as general_err:
+        app.logger.exception(f"Unexpected Python error proxying {target_full_url}")
+        return f"Unexpected server error.", 500
+
+if __name__ == '__main__':
+    port = int(os.environ.get("PORT", 7860)) # پورت استاندارد Hugging Face Spaces
+    # debug_mode = os.environ.get('FLASK_ENV') == 'development' or os.environ.get('FLASK_DEBUG', '0') == '1'
+    # در Hugging Face معمولا FLASK_DEBUG تنظیم نمی‌شود، مگر اینکه خودتان در Dockerfile یا secrets تنظیم کنید
+    debug_mode = os.environ.get('FLASK_DEBUG', '0') == '1'
+    
+    print(f"INFO: Starting Flask app on host 0.0.0.0, port {port}, debug: {debug_mode}")
+    app.run(host='0.0.0.0', port=port, debug=debug_mode)