Update app.py

app.py

@@ -3,8 +3,8 @@ import subprocess
 import os
 
 def run_katana(url):
-    # Set a custom configuration directory
-    katana_config_dir = "/tmp/katana"
+    # Set a custom configuration directory within /app
+    katana_config_dir = "/app/katana"
     os.makedirs(katana_config_dir, exist_ok=True)
     
     # Ensure the config file exists, or create a dummy one if needed
@@ -13,7 +13,7 @@ def run_katana(url):
         with open(config_file_path, 'w') as f:
             f.write("# Dummy configuration\n")
 
-    output_file = "/tmp/urls.txt"  # Use /tmp for writable directory
+    output_file = "/app/urls.txt"  # Use /app for writable directory
     command = f"katana -config {config_file_path} -u {url} > {output_file}"
     try:
         subprocess.run(command, shell=True, check=True)
@@ -21,8 +21,8 @@ def run_katana(url):
     except subprocess.CalledProcessError as e:
         return f"Error running katana: {e}"
 
-# Ensure the /tmp/flagged directory exists
-os.makedirs("/tmp/flagged", exist_ok=True)
+# Ensure the /app/flagged directory exists
+os.makedirs("/app/flagged", exist_ok=True)
 
 # Gradio interface
 interface = gr.Interface(
@@ -31,7 +31,7 @@ interface = gr.Interface(
     outputs="file",
     title="Katana URL Crawler",
     description="Enter a URL to crawl with Katana.",
-    flagging_dir="/tmp/flagged"  # Specify a writable directory for flagging
+    flagging_dir="/app/flagged"  # Specify a writable directory for flagging
 )
 
 interface.launch(server_name="0.0.0.0", server_port=7860)