Update app.py

app.py

@@ -212,10 +212,14 @@ def download_file(choice):
     db_file_path = os.path.join(directory, 'results.db')
 
     if choice == "Database (.db)":
-        # Simply return the path to the database file
+        # Return the correct database file path
         return db_file_path
     
     elif choice == "Database (.html)":
+        # Check if the database file exists
+        if not os.path.isfile(db_file_path):
+            raise FileNotFoundError(f"Database file not found at path: {db_file_path}")
+
         # Connect to the SQLite database
         conn = sqlite3.connect(db_file_path)
         
@@ -223,7 +227,6 @@ def download_file(choice):
             # 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
         
@@ -267,10 +270,9 @@ def download_interface(choice):
         
         # Return the file path (string) directly for the Gradio component to handle
         return file_path
-    except FileNotFoundError as e:
-        return str(e)  # Handle file not found error gracefully
-    except ValueError as e:
-        return str(e)  # Handle any value errors (e.g., invalid choice)
+    except (FileNotFoundError, ValueError) as e:
+        # Display error message in Gradio output
+        return str(e)
 
 # Gradio Blocks
 with gr.Blocks() as demo: