Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import sqlite3
 from PIL import Image
 import os
+import pandas as pd
 
 # Database setup
 DB_PATH = "students.db"
@@ -37,6 +38,27 @@ def register_student(name, age, image_file):
     conn.close()
     return f"Student {name} registered successfully!"
 
+def view_students():
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
+    cursor.execute("SELECT id, name, age, image_path FROM students")
+    records = cursor.fetchall()
+    conn.close()
+
+    # Format data for display
+    if records:
+        data = []
+        for record in records:
+            student_id, name, age, image_path = record
+            if image_path and os.path.exists(image_path):
+                image = Image.open(image_path)
+            else:
+                image = None
+            data.append({"ID": student_id, "Name": name, "Age": age, "Image": image})
+        return pd.DataFrame(data).drop(columns=["Image"]).to_string(), data  # Display text table without images
+    else:
+        return "No records found.", None
+
 # Initialize the database
 setup_database()
 
@@ -44,7 +66,7 @@ setup_database()
 with gr.Blocks() as demo:
     with gr.Row():
         gr.Markdown("### Student Registration")
-    
+
     with gr.Row():
         name_input = gr.Textbox(label="Name", placeholder="Enter the student's name")
         age_input = gr.Number(label="Age", precision=0)
@@ -59,5 +81,22 @@ with gr.Blocks() as demo:
         outputs=output,
     )
 
+    with gr.Row():
+        gr.Markdown("### View Registered Students")
+
+    view_button = gr.Button("View Records")
+    record_output = gr.Textbox(label="Student Records")
+    image_output = gr.Image(label="Images", type="pil", visible=False)  # For displaying images if needed
+
+    def view_records_handler():
+        records, images = view_students()
+        return records
+
+    view_button.click(
+        view_records_handler,
+        inputs=[],
+        outputs=record_output,
+    )
+
 # Launch the Gradio app
 demo.launch()