Update app.py

app.py

@@ -1,49 +1,40 @@
-import gradio as gr
 import subprocess
+import gradio as gr
+import tempfile
 import os
 
-def crawl_website(url):
-  """Crawls a website using Katana and saves the output to a text file."""
-
-  output_file = "katana_output.txt"
-  katana_command = [
-      "katana",
-      "-u", url,
-      "-o", output_file,
-      "-silent"  # Only display output, no progress bars or logs
-  ]
-
-  # Execute Katana command
-  try:
-    subprocess.run(katana_command, check=True)
-  except subprocess.CalledProcessError as e:
-    return f"Error running Katana: {e}", None
-
-  # Read output file content
-  with open(output_file, "r") as f:
-    output_content = f.read()
-
-  return output_content, output_file
-
-with gr.Blocks() as demo:
-  gr.Markdown("## Website Crawler using Katana")
-  with gr.Row():
-    url_input = gr.Textbox(label="Enter Website URL")
-    output_text = gr.Textbox(label="Crawled Endpoints", lines=15)
-  download_button = gr.Button("Download Output File")
-  
-  # Trigger crawling on button click
-  url_input.submit(
-    fn=crawl_website,
-    inputs=url_input,
-    outputs=[output_text, download_button]
-  )
-
-  # Download output file
-  download_button.click(
-    fn=lambda file: file,
-    inputs=download_button,
-    outputs=gr.File(label="Download Results")
-  )
-
-demo.launch()
+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
+    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
+    else:
+        return result, None
+
+iface = gr.Interface(
+    fn=process_and_display,
+    inputs="text",
+    outputs=[
+        gr.Textbox(label="Crawl Results"),
+        gr.File(label="Download Results")
+    ],
+    title="Katana Crawler",
+    description="Enter a URL to crawl using Katana. Results will be displayed and available for download."
+)
+
+iface.launch(server_name="0.0.0.0", server_port=7860)