Update app.py

app.py

@@ -2,26 +2,28 @@ import subprocess
 import gradio as gr
 import tempfile
 import os
-import io
+
 
 def run_katana(url):
-     try:
-        result = subprocess.run(["katana", "-u", url], capture_output=True, text=True, check=True)
-           
-           # Create an in-memory file-like object
-        buffer = io.StringIO(result.stdout)
-           
-           # Generate a filename based on the URL
-        filename = f"katana_results_{url.replace('://', '_').replace('/', '_')}.txt"
-           
-        return result.stdout, (buffer, filename)
-     except Exception as e:
+    try:
+        # Create a temporary file to store the results
+        with tempfile.NamedTemporaryFile(mode='w+', delete=False, suffix='.txt') as temp_file:
+            # Run Katana and save output to the temporary file
+            subprocess.run(["katana", "-u", url, "-o", temp_file.name], check=True)
+
+            # Read the contents of the file
+            temp_file.seek(0)
+            result = temp_file.read()
+
+        # Return the result and the path to the temporary file
+        return result, temp_file.name
+    except Exception as e:
         return str(e), None
 
 def process_and_display(url):
-    result, file_data = run_katana(url)
-    if file_data:
-        return result, file_data
+    result, file_path = run_katana(url)
+    if file_path:
+        return result, file_path
     else:
         return result, None