Update app.py

app.py

@@ -5,21 +5,26 @@ import os
 def run_katana(url):
     katana_config_dir = "/tmp/katana"
     os.makedirs(katana_config_dir, exist_ok=True)
-       
+    
     config_file_path = os.path.join(katana_config_dir, "config.yaml")
-    with open(config_file_path, 'w') as f:
-        f.write("headless: true\n")  # Add a valid configuration option
-   
+    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"
     command = f"katana -config {config_file_path} -u {url} -o {output_file}"
+    
     try:
-        subprocess.run(command, shell=True, check=True, stderr=subprocess.PIPE, text=True)
+        result = subprocess.run(command, shell=True, check=True, capture_output=True, text=True)
         if os.path.exists(output_file):
             return output_file
         else:
-            return "Katana completed but no output file was created."
+            return f"Katana ran successfully, but no output file was created. Katana output: {result.stdout}"
     except subprocess.CalledProcessError as e:
-        return f"Error running katana: {e.stderr}"
+        error_message = f"Error running katana: {e.stderr}"
+        with open(output_file, 'w') as f:
+            f.write(error_message)
+        return output_file
 
 # Ensure the /tmp/flagged directory exists
 os.makedirs("/tmp/flagged", exist_ok=True)