Update app.py

app.py

@@ -14,14 +14,14 @@ def setup_database():
         CREATE TABLE IF NOT EXISTS students (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             name TEXT,
-            rno INTEGER,
+            age INTEGER,
             image_path TEXT
         )
     """)
     conn.commit()
     conn.close()
 
-def register_student(name, rno, image_file):
+def register_student(name, age, image_file):
     conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
 
@@ -32,30 +32,11 @@ def register_student(name, rno, image_file):
     else:
         image_path = None
 
-    cursor.execute("INSERT INTO students (name, rno, image_path) VALUES (?, ?, ?)", (name, rno, image_path))
+    cursor.execute("INSERT INTO students (name, age, image_path) VALUES (?, ?, ?)", (name, age, image_path))
     conn.commit()
     conn.close()
     return f"Student {name} registered successfully!"
 
-def view_records():
-    conn = sqlite3.connect(DB_PATH)
-    cursor = conn.cursor()
-    cursor.execute("SELECT name, rno, image_path FROM students")
-    records = cursor.fetchall()
-    conn.close()
-
-    if not records:
-        return "No records found."
-
-    output = []
-    for name, age, image_path in records:
-        record = f"Name: {name}, ROLL_NO: {rno}"
-        if image_path and os.path.exists(image_path):
-            record += f"\nImage: {image_path}"
-        output.append(record)
-
-    return "\n\n".join(output)
-
 # Initialize the database
 setup_database()
 
@@ -66,27 +47,17 @@ with gr.Blocks() as demo:
     
     with gr.Row():
         name_input = gr.Textbox(label="Name", placeholder="Enter the student's name")
-        rno_input = gr.Number(label="ROLL_NO", precision=0)
+        age_input = gr.Number(label="Age", precision=0)
         image_input = gr.Image(label="Upload Image", type="filepath")
 
-    with gr.Row():
-        register_button = gr.Button("Register Student")
-        record_button = gr.Button("View Records")
-
+    register_button = gr.Button("Register Student")
     output = gr.Textbox(label="Status")
 
     register_button.click(
         register_student,
-        inputs=[name_input, rno_input, image_input],
+        inputs=[name_input, age_input, image_input],
         outputs=output,
     )
 
-    record_button.click(
-        view_records,
-        outputs=output
-    )
-
 # Launch the Gradio app
 demo.launch()
-
-