Fix Session Path: Inject into /root/.local for true resume

app.py

@@ -2,16 +2,33 @@ import gradio as gr
 import subprocess
 import os
 import shutil
+from pathlib import Path
 
 # --- Session Restoration Logic ---
 def restore_session():
-    # If session.sqlite was uploaded to the repo root, move it to the correct path
-    if os.path.exists("session.sqlite"):
-        target_dir = os.path.join("sqlmap-dev", "output", "hashi.ae")
-        os.makedirs(target_dir, exist_ok=True)
+    # SQLMap in Docker uses /root/.local/share/sqlmap/output/
+    # We need to calculate the target directory based on the hostname (hashi.ae)
+    # or just copy to the most likely path.
+    
+    session_source = "session.sqlite"
+    
+    # Target directory used by SQLMap (detected from logs)
+    target_base = Path("/root/.local/share/sqlmap/output/hashi.ae")
+    
+    if os.path.exists(session_source):
         try:
-            shutil.copy("session.sqlite", os.path.join(target_dir, "session.sqlite"))
-            return "✅ Victory Session Restored from Repository."
+            # Create the directory if it doesn't exist
+            target_base.mkdir(parents=True, exist_ok=True)
+            
+            # Copy the session file
+            shutil.copy(session_source, target_base / "session.sqlite")
+            
+            # Also try the www. variant to be safe
+            target_www = Path("/root/.local/share/sqlmap/output/www.hashi.ae")
+            target_www.mkdir(parents=True, exist_ok=True)
+            shutil.copy(session_source, target_www / "session.sqlite")
+            
+            return f"✅ Victory Session Injected into {target_base}"
         except Exception as e:
             return f"⚠️ Session restore warning: {str(e)}"
     return "ℹ️ No session file found in repository."
@@ -25,7 +42,6 @@ def run_sqlmap(url, threads, level, risk, tamper, techn, proxy, extra_args):
         return
 
     # Base command - pointing to the pre-cloned directory
-    # Using absolute path to be safe, or relative if WORKDIR is /app
     cmd = ["python3", "/app/sqlmap-dev/sqlmap.py", "-u", url, "--batch"]
     
     # Performance & Level
@@ -121,6 +137,5 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="blue", secondary_hue="slate"))
     )
 
 if __name__ == "__main__":
-    # Removed runtime cloning logic as it's now handled by Dockerfile
     print("✨ SLMP Panel Live.")
     demo.queue().launch(server_name="0.0.0.0", server_port=7860)