Update app.py

app.py

@@ -203,20 +203,18 @@ def view_database():
 
 # Function to download database or HTML file
 def download_file(choice):
-    directory = tempfile.gettempdir()
-    
-    # Ensure the directory exists
-    if not os.path.exists(directory):
-        os.makedirs(directory)
-
-    db_file_path = os.path.join(directory, 'results.db')
+    db_file_path = 'results.db'  # Directly reference the existing database file
 
     if choice == "Database (.db)":
-        # Return the database file path
-        return db_file_path
-    
+        # 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.")
+
     elif choice == "Database (.html)":
-        # Connect to the SQLite database
+        # Connect to the existing SQLite database
         conn = sqlite3.connect(db_file_path)
         
         try:
@@ -231,7 +229,7 @@ def download_file(choice):
         conn.close()
 
         # Define the path for the HTML file
-        html_file_path = os.path.join(directory, "results.html")
+        html_file_path = os.path.join(tempfile.gettempdir(), "results.html")
         
         # Save the DataFrame as an HTML file
         df.to_html(html_file_path, index=False)