Update app.py

app.py

@@ -2,41 +2,46 @@ from flask import Flask, request, jsonify
 import requests  
 import logging  
 
-# 设置日志记录，便于调试  
 logging.basicConfig(level=logging.INFO)  
 
 app = Flask(__name__)  
 
-def proxy_request(url):  
+def proxy_request(full_url):  
     try:  
-        # 获取原始请求的头部和参数  
-        headers = {key: value for key, value in request.headers.items() if key.lower() != 'host'}  # 避免Host冲突  
+        headers = {key: value for key, value in request.headers.items() if key.lower() != 'host'}  
+        headers['User-Agent'] = request.headers.get('User-Agent', 'Mozilla/5.0 (compatible; ProxyBot/1.0)')  # 添加默认User-Agent  
         response = requests.request(  
             method=request.method,  
-            url=url,  
+            url=full_url,  # 确保是完整的URL  
             headers=headers,  
             data=request.get_data(),  
             params=request.args,  
-            timeout=10  # 设置超时时间，避免卡住  
+            timeout=60,  # 延长超时到60秒，防止网络延迟  
+            verify=True  # 确保SSL验证  
         )  
-        # 返回响应的内容和状态码  
         return response.content, response.status_code, response.headers  
     except requests.exceptions.RequestException as e:  
         logging.error(f"代理请求错误: {e}")  
-        return jsonify({"error": "请求失败，请检查URL或网络连接"}), 500  
+        return jsonify({"error": str(e)}), 500  
 
-@app.route('/proxy/<path:url>')  # 使用<path:url>来处理动态路径  
+@app.route('/proxy/<path:url>')  
 def handle_proxy(url):  
-    full_url = f"https://{url}"  # 假设目标是HTTPS网站，你可以根据需要调整  
+    # 确保url是完整的；如果不是，添加协议  
+    if not url.startswith('http'):  
+        full_url = f"https://{url}"  
+    else:  
+        full_url = url  # 如果用户输入了完整的URL，就直接用  
     content, status_code, headers = proxy_request(full_url)  
     response = app.response_class(  
         response=content,  
         status=status_code,  
-        mimetype='application/octet-stream'  # 默认类型，可以根据响应调整  
+        mimetype=headers.get('Content-Type', 'text/html')  # 自动获取正确的MIME类型  
     )  
+    # 过滤并传递头部，避免冲突  
     for key, value in headers.items():  
-        response.headers[key] = value  # 传递原始响应头部  
+        if key.lower() not in ['transfer-encoding', 'connection']:  # 忽略可能导致502的头部  
+            response.headers[key] = value  
     return response  
 
 if __name__ == '__main__':  
-    app.run(host='0.0.0.0', port=7860, debug=True)  # 在Spaces上，默认端口是7860  
+    app.run(host='0.0.0.0', port=7860, debug=True)