Update app.py

app.py

@@ -1,27 +1,26 @@
-from fastapi import FastAPI, Request
-from fastapi.responses import Response
-import requests
-
-app = FastAPI()
-
-@app.get("/")
-async function handleRequest(request: Request):
-    target_url = request.query_params.get('url')
-    if not target_url:
-        return Response(content="Missing 'url' query parameter", status_code=400)
-
-    try:
-        resp = requests.get(target_url, headers={'User-Agent': 'Mozilla/5.0'})
-        excluded_headers = ['content-encoding', 'content-length', 'transfer-encoding', 'connection']
-        headers = {k: v for k, v in resp.headers.items() if k.lower() not in excluded_headers}
-
-        content_type = resp.headers.get('content-type')
-        if content_type and 'text/html' in content_type:
-            text = resp.text
-            text = text.replace('href="/', f'href="?url={target_url}/')
-            text = text.replace('src="/', f'src="?url={target_url}/')
-            return Response(content=text, headers=headers, status_code=resp.status_code)
-        
-        return Response(content=resp.content, headers=headers, status_code=resp.status_code)
-    except Exception as e:
-        return Response(content=f"Error fetching {target_url}: {str(e)}", status_code=500)
+from fastapi import FastAPI, Request, Response
+import requests
+
+app = FastAPI()
+
+@app.get("/")
+async def proxy(request: Request):
+    target_url = request.query_params.get('url')
+    if not target_url:
+        return Response(content="Missing 'url' query parameter", status_code=400)
+
+    try:
+        resp = requests.get(target_url, headers={'User-Agent': 'Mozilla/5.0'})
+        excluded_headers = ['content-encoding', 'content-length', 'transfer-encoding', 'connection']
+        headers = {k: v for k, v in resp.headers.items() if k.lower() not in excluded_headers}
+
+        content_type = resp.headers.get('content-type')
+        if content_type and 'text/html' in content_type:
+            text = resp.text
+            text = text.replace('href="/', f'href="?url={target_url}/')
+            text = text.replace('src="/', f'src="?url={target_url}/')
+            return Response(content=text, headers=headers, status_code=resp.status_code)
+        
+        return Response(content=resp.content, headers=headers, status_code=resp.status_code)
+    except Exception as e:
+        return Response(content=f"Error fetching {target_url}: {str(e)}", status_code=500)