Update app.py

app.py

@@ -3,23 +3,23 @@ import subprocess
 import os
 
 def run_katana(url):
-    # Set a custom configuration directory within /tmp
     katana_config_dir = "/tmp/katana"
     os.makedirs(katana_config_dir, exist_ok=True)
-    
-    # Ensure the config file exists, or create a dummy one if needed
+       
     config_file_path = os.path.join(katana_config_dir, "config.yaml")
-    if not os.path.exists(config_file_path):
-        with open(config_file_path, 'w') as f:
-            f.write("# Dummy configuration\n")
-
-    output_file = "/tmp/urls.txt"  # Use /tmp for writable directory
-    command = f"katana -config {config_file_path} -u {url} > {output_file}"
+    with open(config_file_path, 'w') as f:
+        f.write("headless: true\n")  # Add a valid configuration option
+   
+    output_file = "/tmp/urls.txt"
+    command = f"katana -config {config_file_path} -u {url} -o {output_file}"
     try:
-        subprocess.run(command, shell=True, check=True)
-        return output_file
+        subprocess.run(command, shell=True, check=True, stderr=subprocess.PIPE, text=True)
+        if os.path.exists(output_file):
+            return output_file
+        else:
+            return "Katana completed but no output file was created."
     except subprocess.CalledProcessError as e:
-        return f"Error running katana: {e}"
+        return f"Error running katana: {e.stderr}"
 
 # Ensure the /tmp/flagged directory exists
 os.makedirs("/tmp/flagged", exist_ok=True)