Update app.py

app.py

@@ -26,19 +26,23 @@ class MirrorScraper:
         homepage_url = f"{self.base_url}/"
         
         logger.info("🔄 Extracting CSRF token from homepage...")
-        response = requests.get(homepage_url, headers=self.headers)
-        
-        if response.status_code == 200:
-            csrf_match = re.search(r'name="csrf"[^>]*value="([^"]+)"', response.text)
-            if csrf_match:
-                csrf_token = csrf_match.group(1)
-                logger.info("✅ CSRF token extracted from homepage")
-                return csrf_token
+        try:
+            response = requests.get(homepage_url, headers=self.headers, timeout=10)
+            
+            if response.status_code == 200:
+                csrf_match = re.search(r'name="csrf"[^>]*value="([^"]+)"', response.text)
+                if csrf_match:
+                    csrf_token = csrf_match.group(1)
+                    logger.info("✅ CSRF token extracted from homepage")
+                    return csrf_token
+                else:
+                    logger.error("❌ CSRF token not found in homepage")
+                    return None
             else:
-                logger.error("❌ CSRF token not found in homepage")
+                logger.error(f"❌ Failed to get homepage. Status: {response.status_code}")
                 return None
-        else:
-            logger.error(f"❌ Failed to get homepage. Status: {response.status_code}")
+        except Exception as e:
+            logger.error(f"❌ Error extracting CSRF: {str(e)}")
             return None
 
     def get_server_data_and_csrf(self) -> tuple:
@@ -63,24 +67,28 @@ class MirrorScraper:
         }
         
         logger.info("🔄 Getting server data and new CSRF token...")
-        response = requests.post(servers_url, data=data, headers=headers)
-        
-        if response.status_code == 200:
-            # Extract NEW CSRF token from the JavaScript data
-            csrf_match = re.search(r'data-csrf=""([^&]+)""', response.text)
-            if csrf_match:
-                new_csrf = csrf_match.group(1)
-                logger.info("✅ New CSRF token extracted")
-            else:
-                logger.warning("⚠️ New CSRF token not found, using fallback")
-                new_csrf = "RXFOUlV0ZjFxdHg5L1VGd1ZrazRKVFRDUGdNQ1ZhUG01N2l3ZitXTFFMUU81Z2tUb1JXY3JoOXJveWtaSHA2clNBRHZhSFNIN1N0Rk8rak13c3dJNFBEcEcwa3VjYndCMVo5emEyd0ZUQ2RzK0U3RCtNR3dYcGVVSkxTamx6bzl4M2orNkN0Q2JwLzZHU0F6NHkwSTlRPT0="
-            
-            # Extract server ID (use first one from the list or fallback)
-            server_id = "161"  # Fallback to known working server
+        try:
+            response = requests.post(servers_url, data=data, headers=headers, timeout=10)
             
-            return server_id, new_csrf
-        else:
-            logger.error(f"❌ Failed to get servers page. Status: {response.status_code}")
+            if response.status_code == 200:
+                # Extract NEW CSRF token from the JavaScript data
+                csrf_match = re.search(r'data-csrf=""([^&]+)""', response.text)
+                if csrf_match:
+                    new_csrf = csrf_match.group(1)
+                    logger.info("✅ New CSRF token extracted")
+                else:
+                    logger.warning("⚠️ New CSRF token not found, using fallback")
+                    new_csrf = "RXFOUlV0ZjFxdHg5L1VGd1ZrazRKVFRDUGdNQ1ZhUG01N2l3ZitXTFFMUU81Z2tUb1JXY3JoOXJveWtaSHA2clNBRHZhSFNIN1N0Rk8rak13c3dJNFBEcEcwa3VjYndCMVo5emEyd0ZUQ2RzK0U3RCtNR3dYcGVVSkxTamx6bzl4M2orNkN0Q2JwLzZHU0F6NHkwSTlRPT0="
+                
+                # Extract server ID (use first one from the list or fallback)
+                server_id = "161"  # Fallback to known working server
+                
+                return server_id, new_csrf
+            else:
+                logger.error(f"❌ Failed to get servers page. Status: {response.status_code}")
+                return None, None
+        except Exception as e:
+            logger.error(f"❌ Error getting server data: {str(e)}")
             return None, None
 
     def get_complete_url(self) -> Dict[str, str]:
@@ -89,7 +97,7 @@ class MirrorScraper:
             # Get server ID and NEW CSRF token
             server_id, new_csrf = self.get_server_data_and_csrf()
             if not server_id or not new_csrf:
-                return {"error": "Cannot proceed without server data"}
+                return {"success": False, "error": "Cannot proceed without server data"}
             
             # Use the NEW CSRF token for the final request
             url = f"{self.base_url}/requests?fso="
@@ -110,7 +118,7 @@ class MirrorScraper:
             }
             
             logger.info("🚀 Getting complete URL from mirror site...")
-            response = requests.post(url, data=data, headers=headers, allow_redirects=False)
+            response = requests.post(url, data=data, headers=headers, allow_redirects=False, timeout=10)
             
             logger.info(f"📊 Status: {response.status_code}")
             
@@ -128,11 +136,11 @@ class MirrorScraper:
                 if 'location' not in response.headers:
                     error_msg += " - No location header in response"
                 logger.error(f"❌ {error_msg}")
-                return {"error": error_msg}
+                return {"success": False, "error": error_msg}
                 
         except Exception as e:
             logger.error(f"❌ Exception occurred: {str(e)}")
-            return {"error": f"Internal server error: {str(e)}"}
+            return {"success": False, "error": f"Internal server error: {str(e)}"}
 
 # Initialize the scraper
 scraper = MirrorScraper()
@@ -140,17 +148,44 @@ scraper = MirrorScraper()
 @app.route('/')
 def home():
     return jsonify({
-        "message": "Mirror URL Generator API",
+        "message": "Mirror URL Generator API - huijio/urltemplate",
         "status": "active",
+        "version": "1.0",
         "endpoints": {
             "/generate": "GET - Generate a new URL",
-            "/health": "GET - Health check"
-        }
+            "/health": "GET - Health check",
+            "/docs": "GET - API documentation"
+        },
+        "example_usage": "GET https://huijio-urltemplate.hf.space/generate"
     })
 
 @app.route('/health')
 def health():
-    return jsonify({"status": "healthy"})
+    return jsonify({"status": "healthy", "service": "mirror-url-generator"})
+
+@app.route('/docs')
+def docs():
+    return jsonify({
+        "api_documentation": {
+            "base_url": "https://huijio-urltemplate.hf.space",
+            "endpoints": {
+                "/generate": {
+                    "method": "GET",
+                    "description": "Generate a new proxy URL",
+                    "response": {
+                        "success": "boolean",
+                        "url": "string (the generated URL)",
+                        "message": "string"
+                    }
+                },
+                "/health": {
+                    "method": "GET", 
+                    "description": "Health check endpoint",
+                    "response": {"status": "string"}
+                }
+            }
+        }
+    })
 
 @app.route('/generate', methods=['GET'])
 def generate_url():
@@ -159,7 +194,10 @@ def generate_url():
         result = scraper.get_complete_url()
         return jsonify(result)
     except Exception as e:
-        return jsonify({"error": f"Internal server error: {str(e)}"}), 500
+        return jsonify({
+            "success": False, 
+            "error": f"Internal server error: {str(e)}"
+        }), 500
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=False)