Update app.py

app.py

@@ -1,21 +1,26 @@
 import gradio as gr
 import subprocess
 import piper
+import os
 
-def run_command(command):
- try:
- # Running the command and capturing the output
- result = subprocess.run(command, shell=True, text=True, capture_output=True, check=True)
- return result.stdout
- except subprocess.CalledProcessError as e:
- return e.stderr
+def run_command_and_get_file(command):
+ # Run the command
+ subprocess.run(command, shell=True, text=True, capture_output=True)
+
+ # Assuming the command creates a file with a known name
+ file_path = 'output.txt' # Replace with the actual expected file path
+
+ if os.path.exists(file_path):
+ return file_path
+ else:
+ return "No file generated."
 
 iface = gr.Interface(
- fn=run_command,
+ fn=run_command_and_get_file,
  inputs='text',
- outputs='text',
+ outputs='file', # Output type changed to 'file'
  title="CLI Interface",
- description="Enter your command and see the output."
+ description="Enter your command. If a file is generated, you can download it."
 )
 
-iface.launch()
+iface.launch()