Update app.py

app.py

@@ -205,20 +205,16 @@ def download_file(choice):
     if choice == "Database (.db)":
         # Provide the path to the database file
         return 'results.db'
-    else:
+    elif choice == "Database (.html)":
         conn = sqlite3.connect('results.db')
-        c = conn.cursor()
-        c.execute("SELECT predicted_image FROM results ORDER BY id DESC LIMIT 1")
-        row = c.fetchone()
+        df = pd.read_sql_query("SELECT * FROM results", conn)
         conn.close()
-        if row:
-            image_bytes = row[0]
-            with tempfile.NamedTemporaryFile(delete=False, suffix='.png') as temp_file:
-                temp_file.write(image_bytes)
-                temp_file.flush()  # Ensure all data is written before closing
-                return temp_file.name
-        else:
-            raise FileNotFoundError("No images found in the database.")
+
+        # Save the DataFrame as an HTML file
+        html_file_path = "/mnt/data/results.html"
+        df.to_html(html_file_path, index=False)
+
+        return html_file_path
 
 # Initialize the database
 init_db()
@@ -252,7 +248,6 @@ with gr.Blocks() as demo:
     with gr.Column():
         gr.Markdown("# Cataract Detection System")
         gr.Markdown("Upload an image to detect cataract and add patient details.")
-        gr.Image("PR_curve.png", label="Model PR Curve")
         gr.Markdown("This application uses YOLOv8 with mAP=0.981")
     
     with gr.Column():
@@ -279,7 +274,7 @@ with gr.Blocks() as demo:
     view_db_btn.click(fn=view_db_interface, inputs=[], outputs=[db_output])
 
     with gr.Column():
-        download_choice = gr.Radio(["Database (.db)", "Predicted Image (.png)"], label="Choose the file to download:")
+        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")