Update main.py

main.py

@@ -2,7 +2,7 @@ import requests
 import random
 import time
 import threading
-from flask import Flask, request, Response
+from flask import Flask, request, Response, jsonify
 
 PROXY_LIST_URL = "https://proxies.typegpt.net/ips.txt"
 proxies_cache = []
@@ -11,7 +11,6 @@ last_refresh = 0
 app = Flask(__name__)
 
 def fetch_proxies():
-    """Fetch proxy list from remote URL."""
     try:
         resp = requests.get(PROXY_LIST_URL, timeout=10)
         resp.raise_for_status()
@@ -22,13 +21,11 @@ def fetch_proxies():
         return []
 
 def get_random_proxy(proxies):
-    """Pick a random proxy from the list."""
     if not proxies:
         return None
     return random.choice(proxies)
 
 def refresh_proxies_loop():
-    """Background thread: refresh proxy list every 5 minutes."""
     global proxies_cache, last_refresh
     while True:
         if time.time() - last_refresh > 300 or not proxies_cache:
@@ -43,33 +40,33 @@ def health():
 
 @app.route("/<path:target_url>")
 def proxy_request(target_url):
-    """Catch-all route: forward request to target_url using random proxy."""
     if not target_url.startswith("http"):
         target_url = "http://" + target_url  # allow shorthand
 
     proxy = get_random_proxy(proxies_cache)
     if not proxy:
-        return {"error": "No proxies available"}, 500
+        return jsonify({"error": "No proxies available"}), 500
 
     proxies = {"http": proxy, "https": proxy}
     try:
         print(f"[INFO] Forwarding to {target_url} via {proxy}")
         resp = requests.get(target_url, proxies=proxies, timeout=15)
 
+        # Add "X-Proxy-Used" header so user knows which IP was used
+        headers = dict(resp.headers)
+        headers["X-Proxy-Used"] = proxy
+
         return Response(
             resp.content,
             status=resp.status_code,
-            headers=dict(resp.headers),
+            headers=headers,
         )
     except Exception as e:
-        return {"error": f"Proxy failed: {proxy}", "details": str(e)}, 502
+        return jsonify({"error": f"Proxy failed", "proxy": proxy, "details": str(e)}), 502
 
 def main():
-    # Start proxy refresher thread
     t = threading.Thread(target=refresh_proxies_loop, daemon=True)
     t.start()
-
-    # Run Flask server
     app.run(host="0.0.0.0", port=5000)
 
 if __name__ == "__main__":