Update main.py

main.py

@@ -3,47 +3,45 @@ import requests
 
 app = Flask(__name__)
 
-# 目标网站的基础URL，修改为你想要代理的目标网站
-BASE_URL = 'https://chatgpt.vvqq.top'
-
-@app.route('/', defaults={'path': ''}, methods=['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'HEAD', 'OPTIONS'])
-@app.route('/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE', 'PATCH', 'HEAD', 'OPTIONS'])
+PROXY_TARGET = "https://chatgpt.vvqq.top/"
+
+@app.route('/')
+def home():
+    global PROXY_TARGET
+    # 获取原始HTTP GET请求的参数
+    resp = requests.get(f"{PROXY_TARGET}{request.full_path}")
+    # 返回从目标网站获取的响应数据
+    excluded_headers = ['content-encoding', 'content-length', 'transfer-encoding', 'connection']
+    headers = [(name, value) for (name, value) in resp.raw.headers.items()
+               if name.lower() not in excluded_headers]
+    response = Response(resp.content, resp.status_code, headers)
+    return response
+
+@app.route('/<path:path>', methods=['GET', 'POST', 'PUT', 'DELETE'])
 def proxy(path):
-    # 构建新的请求 URL，将原始请求的路径和查询参数附加到目标基础URL后
-    target_url = f"{BASE_URL}/{path}"
-    if request.query_string:
-        target_url += f"?{request.query_string.decode('utf-8')}"
-
-    print(f"Target URL: {target_url}")
-
-    # 复制原始请求的头部信息，以便在新请求中使用
-    headers = {key: value for key, value in request.headers if key != 'Host'}
-    print(f"Headers: {headers}")
-
-    # 如果存在，复制原始请求的正文
-    body = request.get_data() if request.method not in ["GET", "HEAD"] else None
-    print(f"Body: {body}")
-
-    try:
-        # 使用 requests 发送新请求到目标网站
-        response = requests.request(
-            method=request.method,
-            url=target_url,
-            headers=headers,
-            data=body,
-            allow_redirects=True
-        )
-
-        # 构建新的响应
-        proxy_response = Response(response.content, response.status_code)
-        for key, value in response.headers.items():
-            proxy_response.headers[key] = value
-
-        return proxy_response
-
-    except requests.exceptions.RequestException as e:
-        print(f"Request failed: {e}")
-        return Response("Bad Gateway", status=502)
+    global PROXY_TARGET
+    # 构造目标URL
+    url = f"{PROXY_TARGET}/{path}"
+    # 根据原始请求的方法来发送对应的请求到目标服务器
+    if request.method == 'GET':
+        resp = requests.get(url, stream=True)
+    elif request.method == 'POST':
+        resp = requests.post(url, json=request.json, stream=True)
+    elif request.method == 'PUT':
+        resp = requests.put(url, json=request.json, stream=True)
+    elif request.method == 'DELETE':
+        resp = requests.delete(url, stream=True)
+    
+    # 将从目标网站获取的响应返回给客户端
+    excluded_headers = ['content-encoding', 'content-length', 'transfer-encoding', 'connection']
+    headers = [(name, value) for (name, value) in resp.raw.headers.items()
+               if name.lower() not in excluded_headers]
+    
+    def generate():
+        for chunk in resp.iter_content(chunk_size=8192):
+            yield chunk
+    
+    return Response(generate(), resp.status_code, headers)
 
 if __name__ == '__main__':
-    app.run(host='0.0.0.0', port=7860)
+    app.run(host='0.0.0.0', port=7860, debug=True)