Update app.py

app.py

@@ -2,27 +2,26 @@ import subprocess
 import gradio as gr
 import tempfile
 import os
+import io
 
 def run_katana(url):
-    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
+     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:
         return str(e), None
 
 def process_and_display(url):
-    result, file_path = run_katana(url)
-    if file_path:
-        return result, file_path
+    result, file_data = run_katana(url)
+    if file_data:
+        return result, file_data
     else:
         return result, None