Update app.py

app.py

@@ -203,18 +203,20 @@ def view_database():
 
 # Function to download database or HTML file
 def download_file(choice):
-    db_file_path = 'results.db'  # Directly reference the existing database file
+    directory = tempfile.gettempdir()
+    
+    # Ensure the directory exists
+    if not os.path.exists(directory):
+        os.makedirs(directory)
 
-    if choice == "Database (.db)":
-        # Check if the database file exists
-        if os.path.exists(db_file_path):
-            # Return the database file path directly
-            return db_file_path
-        else:
-            raise FileNotFoundError(f"{db_file_path} not found.")
+    db_file_path = os.path.join(directory, 'results.db')
 
+    if choice == "Database (.db)":
+        # Simply return the path to the database file
+        return db_file_path
+    
     elif choice == "Database (.html)":
-        # Connect to the existing SQLite database
+        # Connect to the SQLite database
         conn = sqlite3.connect(db_file_path)
         
         try:
@@ -229,7 +231,7 @@ def download_file(choice):
         conn.close()
 
         # Define the path for the HTML file
-        html_file_path = os.path.join(tempfile.gettempdir(), "results.html")
+        html_file_path = os.path.join(directory, "results.html")
         
         # Save the DataFrame as an HTML file
         df.to_html(html_file_path, index=False)
@@ -263,15 +265,12 @@ def download_interface(choice):
         # Get the file path
         file_path = download_file(choice)
         
-        # Open the file in binary mode and return its content along with the filename
-        with open(file_path, "rb") as file:
-            file_content = file.read()
-            file_name = os.path.basename(file_path)
-            return file_content, file_name  # Return as a tuple for Gradio file component
+        # Return the file path (string) directly for the Gradio component to handle
+        return file_path
     except FileNotFoundError as e:
-        return str(e), None  # Handle file not found error gracefully
+        return str(e)  # Handle file not found error gracefully
     except ValueError as e:
-        return str(e), None  # Handle any value errors (e.g., invalid choice)
+        return str(e)  # Handle any value errors (e.g., invalid choice)
 
 # Gradio Blocks
 with gr.Blocks() as demo:
@@ -308,7 +307,7 @@ with gr.Blocks() as demo:
         download_btn = gr.Button("Download")
         download_output = gr.File(label="Download File")
 
-    download_btn.click(fn=download_interface, inputs=[download_choice], outputs=[download_output])
+    download_btn.click(fn=download_interface, inputs=[download_choice], outputs=gr.File())
 
 # Launch the Gradio app
 demo.launch()