Update app.py

app.py

@@ -183,6 +183,22 @@ def split_and_search(text:str):
         time.sleep(1)
     return all_data
 
+#For View:
+def get_all_plants() -> List[Dict]:
+    """Retrieve all plants from database"""
+    try:
+        conn = sqlite3.connect(DB_NAME)
+        conn.row_factory = sqlite3.Row
+        cursor = conn.cursor()
+        cursor.execute("SELECT `_rowid_`,* FROM plants ORDER BY `_rowid_` DESC")
+        plants = [dict(row) for row in cursor.fetchall()]
+        conn.close()
+        return plants
+    except Exception as e:
+        print(f"Database retrieval error: {e}")
+        return [{"Error": "Failed to retrieve data from database"}]
+
+
 #use it here :
 #process_plants(plants)
 
@@ -206,5 +222,23 @@ with gr.Blocks(title="AI-Powered Medicinal Plants Database") as app:
             outputs=output_area #json_output
         )
 
+    with gr.Tab("View Database"):
+        gr.Markdown("### Stored Plant Information")
+        with gr.Row():
+            refresh_btn = gr.Button("Refresh Data", variant="secondary")
+            clear_db = gr.Button("Clear Database", variant="stop")
+        
+        db_table = gr.Dataframe(
+            headers=["id", "name", "scientific_name", "description"],
+            datatype=["number", "str", "str", "str"],
+            col_count=(4, "fixed"),
+            interactive=True
+        )
+        
+        refresh_btn.click(
+            fn=get_all_plants,
+            outputs=db_table
+        )
+
 if __name__ == "__main__":
     app.launch(debug=True, share=False)