Update main.py

main.py

@@ -6,11 +6,12 @@ import importlib.util
 import uvicorn  # Programmatic bootstrapper
 
 # =====================================================================
-# 1. CRITICAL PATH INJECTION FIX
+# 1. CRITICAL PATH INJECTION FIX (Matched to Root Repository Layout)
 # =====================================================================
 CURRENT_BASE_DIR = os.path.dirname(os.path.abspath(__file__))
 VERSION_1_PATH = os.path.join(CURRENT_BASE_DIR, "Version_1")
-VERSION_2_PATH = os.path.join(CURRENT_BASE_DIR, "Version_2-Copy")
+# 🔥 FIXED: Added spaces around the hyphen to match your exact directory name
+VERSION_2_PATH = os.path.join(CURRENT_BASE_DIR, "Version_2 - Copy")
 VERSION_3_PATH = os.path.join(CURRENT_BASE_DIR, "Version_3")
 VERSION_4_PATH = os.path.join(CURRENT_BASE_DIR, "Version_4")
 VERSION_5_PATH = os.path.join(CURRENT_BASE_DIR, "Version_5")
@@ -70,7 +71,7 @@ except Exception as e:
     print(f"[-] Could not merge Version 1. Error: {e}")
 
 # =====================================================================
-# [ ENGINE 2 ] VERSION 2: VISUAL RESNET18 PHISHING DETECTOR (Dynamic Hyphen Import)
+# [ ENGINE 2 ] VERSION 2: VISUAL RESNET18 PHISHING DETECTOR
 # =====================================================================
 v2_router = APIRouter(prefix="/api/v2", tags=["Version 2: Visual ResNet18 Engine"])
 
@@ -84,7 +85,6 @@ try:
     print("[*] Attempting to load Version 2 visual engine...")
     v2_main_path = os.path.join(VERSION_2_PATH, "main.py")
     
-    # Dynamically load the module to bypass Python's hyphen import restriction
     spec = importlib.util.spec_from_file_location("v2_main", v2_main_path)
     v2_main = importlib.util.module_from_spec(spec)
     sys.modules["v2_main"] = v2_main
@@ -110,13 +110,11 @@ async def analyze_url_vision(payload: V2VisionRequest):
     temp_img_path = os.path.join(OUTPUT_DIR, f"v2_infer_{int(time.time()*1000)}.png")
 
     try:
-        # Step 1: Capture the visual fingerprint
         screenshot_file = await v2_capture_screenshot(url, temp_img_path)
 
         if not screenshot_file or not os.path.exists(screenshot_file):
             raise HTTPException(status_code=502, detail="Failed to capture screenshot. The target may be offline.")
 
-        # Step 2: Run PyTorch Inference
         result = v2_analyzer.analyze_image(screenshot_file)
         latency_ms = (time.perf_counter() - start_time) * 1000
 
@@ -133,7 +131,6 @@ async def analyze_url_vision(payload: V2VisionRequest):
             "latency_ms": round(latency_ms, 2)
         }
     finally:
-        # Crucial Step 3: Prevent disk bloat
         if os.path.exists(temp_img_path):
             os.remove(temp_img_path)
 
@@ -164,7 +161,7 @@ if os.path.exists(MODEL_PATH):
     bst.load_model(MODEL_PATH)
 else:
     bst = None
-    print("[-] Warning: models/final_model.json not found. V4 engine will return a configuration error.")
+    print(f"[-] Warning: {MODEL_PATH} not found. V4 engine will return a configuration error.")
 
 @v4_router.post("/evaluate")
 async def evaluate_xgboost_url(payload: XGBoostRequest):
@@ -347,10 +344,11 @@ if __name__ == "__main__":
     print("[*] Resolving path mapping vectors securely...")
     print(f"[+] Directing Uvicorn to look inside application anchor: {CURRENT_BASE_DIR}")
     
+    # 🔥 FIXED: Points directly to port 7860 for Hugging Face standard routing context
     uvicorn.run(
         "main:app",
         host="0.0.0.0",
-        port=8000,
+        port=7860,
         reload=True,
-        app_dir=CURRENT_BASE_DIR  # Force Uvicorn to anchor right here permanently!
+        app_dir=CURRENT_BASE_DIR
     )