update gitignore for model files

.gitignore

@@ -0,0 +1,3 @@
+models/
+*.pkl
+model_cache.pkl

app.py

@@ -9,6 +9,11 @@ Implementation: Private (using pre-trained models)
 """
 
 # app.py - Main Gradio interface (PUBLIC)
+
+# Add at the top of app.py
+import requests
+import os
+   
 import gradio as gr
 import pickle
 import numpy as np
@@ -20,13 +25,49 @@ import warnings
 warnings.filterwarnings('ignore')
 
 # Load pre-trained detector (implementation hidden in pickle)
+
+def download_model():
+    """Download model from Google Drive if not cached."""
+    model_path = "models/ai_security_detector.pkl"
+    
+    if not os.path.exists(model_path):
+        print("📥 Downloading AI security model...")
+        
+        # Your Google Drive file ID
+        file_id = "1prYcoojiqSpdNzcSm0frrloA3L9lnVC7"
+        download_url = f"https://drive.google.com/uc?id={file_id}"
+        
+        os.makedirs("models", exist_ok=True)
+        
+        try:
+            response = requests.get(download_url)
+            response.raise_for_status()  # Raise exception for bad status codes
+            
+            with open(model_path, "wb") as f:
+                f.write(response.content)
+            
+            print("✅ Model downloaded successfully!")
+            
+        except Exception as e:
+            print(f"❌ Error downloading model: {e}")
+            print("⚠️ Using fallback detector...")
+            return None
+    
+    return model_path
+
+# Load model at startup
 try:
-    with open('models/ai_security_detector.pkl', 'rb') as f:
-        detector = pickle.load(f)
-    print("✅ AI Security Detector loaded successfully")
+    model_path = download_model()
+    if model_path and os.path.exists(model_path):
+        with open(model_path, 'rb') as f:
+            detector = pickle.load(f)
+        print("✅ AI Security Detector loaded successfully")
+    else:
+        detector = None
+        print("⚠️ Using fallback mode")
 except Exception as e:
-    print(f"❌ Error loading detector: {e}")
     detector = None
+    print(f"❌ Error loading detector: {e}")
 
 # Demo configuration
 DEMO_CONFIG = {
@@ -333,4 +374,4 @@ if __name__ == "__main__":
         show_error=True,
         show_tips=True,
         enable_queue=True
-    )
+    )

requirements.txt

@@ -3,4 +3,5 @@ numpy==1.24.3
 pandas==2.0.3
 scikit-learn==1.3.0
 sentence-transformers==2.2.2
-scipy==1.11.1
+scipy==1.11.1
+requests==2.31.0