Update app.py

app.py

@@ -202,6 +202,11 @@ def view_database():
 
 # Function to download database or image
 def download_file(choice):
+    # Ensure the directory exists
+    directory = "/mnt/data"
+    if not os.path.exists(directory):
+        os.makedirs(directory)
+
     if choice == "Database (.db)":
         # Provide the path to the database file
         return 'results.db'
@@ -211,7 +216,7 @@ def download_file(choice):
         conn.close()
 
         # Save the DataFrame as an HTML file
-        html_file_path = "/mnt/data/results.html"
+        html_file_path = os.path.join(directory, "results.html")
         df.to_html(html_file_path, index=False)
 
         return html_file_path