Update app.py

app.py

@@ -209,17 +209,23 @@ def download_file(choice):
     if not os.path.exists(directory):
         os.makedirs(directory)
 
+    db_file_path = os.path.join(directory, 'results.db')
+
     if choice == "Database (.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
+        # Return the database file path
+        return db_file_path
     
     elif choice == "Database (.html)":
         # Connect to the SQLite database
-        conn = sqlite3.connect(os.path.join(directory, 'results.db'))
+        conn = sqlite3.connect(db_file_path)
         
-        # Read the database into a DataFrame
-        df = pd.read_sql_query("SELECT * FROM results", conn)
+        try:
+            # Attempt to read the results table into a DataFrame
+            df = pd.read_sql_query("SELECT * FROM results", conn)
+        except pd.errors.DatabaseError as e:
+            # Handle the case where the table doesn't exist
+            conn.close()
+            raise ValueError("Table 'results' does not exist in the database.") from e
         
         # Close the database connection
         conn.close()
@@ -230,8 +236,9 @@ def download_file(choice):
         # Save the DataFrame as an HTML file
         df.to_html(html_file_path, index=False)
 
-        return html_file_path  # Return the path as a string
-
+        # Return the path to the HTML file
+        return html_file_path
+        
 # Initialize the database
 init_db()