Update app.py

app.py

@@ -201,23 +201,36 @@ def view_database():
 
     return html_content
 
-# Function to download database or image
+# Function to download database or HTML file
 def download_file(choice):
     directory = "/mnt/data"
+    
+    # Ensure the directory exists
     if not os.path.exists(directory):
         os.makedirs(directory)
 
     if choice == "Database (.db)":
-        return 'results.db'
+        # Assuming the database is stored in the same directory
+        db_file_path = os.path.join(directory, 'results.db')
+        return db_file_path  # Return the path as a string
+    
     elif choice == "Database (.html)":
-        conn = sqlite3.connect('results.db')
+        # Connect to the SQLite database
+        conn = sqlite3.connect(os.path.join(directory, 'results.db'))
+        
+        # Read the database into a DataFrame
         df = pd.read_sql_query("SELECT * FROM results", conn)
+        
+        # Close the database connection
         conn.close()
 
+        # Define the path for the HTML file
         html_file_path = os.path.join(directory, "results.html")
+        
+        # Save the DataFrame as an HTML file
         df.to_html(html_file_path, index=False)
 
-        return html_file_path  # Ensure the path is returned as a string
+        return html_file_path  # Return the path as a string
 
 # Initialize the database
 init_db()