Update app.py

app.py

@@ -203,29 +203,44 @@ def view_database():
     return html_content
 
 # Function to download database or HTML file
-def download_file(choice):
-    # Define a temporary directory to store the database file
+def download_file(file_format):
     directory = tempfile.gettempdir()
     db_file_path = os.path.join(directory, 'results.db')
-
-    if choice == "Database (.db)":
-        # Ensure the database file exists
+    
+    if file_format == "Database (.db)":
+        # Simulate creating or writing to the database file here
         if not os.path.exists(db_file_path):
-            # Create and populate the database with dummy data if needed
+            # Create a dummy database file for demonstration
             conn = sqlite3.connect(db_file_path)
-            # Example table creation (if not already exists)
-            conn.execute('''CREATE TABLE IF NOT EXISTS results (
-                            id INTEGER PRIMARY KEY,
-                            name TEXT NOT NULL,
-                            value TEXT NOT NULL
-                        );''')
-            conn.execute("INSERT INTO results (name, value) VALUES ('example', '123')")
+            conn.execute('CREATE TABLE IF NOT EXISTS results (id INTEGER PRIMARY KEY, name TEXT)')
             conn.commit()
             conn.close()
-
-        # Return the database file path for download
         return db_file_path
     
+    elif file_format == "HTML (.html)":
+        # Connect to the SQLite database
+        conn = sqlite3.connect(db_file_path)
+        
+        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:
+            conn.close()
+            raise ValueError("Table 'results' does not exist in the database.") from e
+        
+        # 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
+    else:
+        raise ValueError("Invalid file format specified.")
+    
     elif choice == "Database (.html)":
         # Connect to the SQLite database
         conn = sqlite3.connect(db_file_path)
@@ -314,10 +329,12 @@ with gr.Blocks() as demo:
 
     with gr.Column():
         download_choice = gr.Radio(["Database (.db)", "Database (.html)"], label="Choose the file to download:")
-        download_btn = gr.Button("Download")
-        download_output = gr.File(label="Download File")
+        download_btn_db = gr.Button("Download Database (.db)")
+        download_btn_html = gr.Button("Download HTML (.html)")
+        download_file_output = gr.File(label="Download File")
 
-    download_btn.click(download_file, inputs=choice, outputs=download_file_output)
+    download_btn_db.click(lambda: download_file("Database (.db)"), outputs=download_file_output)
+    download_btn_html.click(lambda: download_file("HTML (.html)"), outputs=download_file_output)
 
 # Launch the Gradio app
 demo.launch()