Update app.py

app.py

@@ -3,7 +3,7 @@ import os
 import numpy as np
 from cataract import combined_prediction, save_cataract_prediction_to_db, predict_object_detection
 from glaucoma import combined_prediction_glaucoma, submit_to_db, predict_image
-from database import get_db_data, format_db_data
+from database import get_db_data, format_db_data, clear_database
 from chatbot import chatbot, update_patient_history, generate_voice_response 
 from PIL import Image
 
@@ -220,4 +220,20 @@ with gr.Blocks(theme=theme) as demo:
         load_db_btn = gr.Button("Load Database Content")
         load_db_btn.click(display_db_data, outputs=db_display)
 
+        # Buttons to clear databases
+        clear_cataract_db_btn = gr.Button("Clear Cataract Database")
+        clear_glaucoma_db_btn = gr.Button("Clear Glaucoma Database")
+
+        clear_cataract_db_btn.click(
+            fn=clear_database,
+            inputs=[gr.State(value=db_path_cataract), gr.State(value="cataract_results")],
+            outputs=db_display
+        )
+
+        clear_glaucoma_db_btn.click(
+            fn=clear_database,
+            inputs=[gr.State(value=db_path_glaucoma), gr.State(value="results")],
+            outputs=db_display
+        )
+        
     demo.launch()