Update app.py

app.py

@@ -1,71 +1,63 @@
 import gradio as gr
 import sqlite3
+from PIL import Image
 import os
 
-# Initialize SQLite database
-def init_db():
-    conn = sqlite3.connect("students_with_images.db")
+# Database setup
+DB_PATH = "students.db"
+os.makedirs("uploads", exist_ok=True)
+
+def setup_database():
+    conn = sqlite3.connect(DB_PATH)
     cursor = conn.cursor()
-    cursor.execute('''
+    cursor.execute("""
         CREATE TABLE IF NOT EXISTS students (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
-            name TEXT NOT NULL,
-            email TEXT NOT NULL UNIQUE,
-            phone TEXT NOT NULL,
-            course TEXT NOT NULL,
-            image_path TEXT NOT NULL
+            name TEXT,
+            age INTEGER,
+            image_path TEXT
         )
-    ''')
+    """)
     conn.commit()
     conn.close()
 
-# Function to handle student registration
-def register_student(name, email, phone, course, image):
-    try:
-        # Save the image
-        image_path = f"student_images/{email.replace('@', '_').replace('.', '_')}.png"
-        os.makedirs("student_images", exist_ok=True)
-        image.save(image_path)
+def register_student(name, age, image_file):
+    conn = sqlite3.connect(DB_PATH)
+    cursor = conn.cursor()
 
-        # Insert data into database
-        conn = sqlite3.connect("students_with_images.db")
-        cursor = conn.cursor()
-        cursor.execute('''
-            INSERT INTO students (name, email, phone, course, image_path)
-            VALUES (?, ?, ?, ?, ?)
-        ''', (name, email, phone, course, image_path))
-        conn.commit()
-        conn.close()
+    # Save image to the uploads directory
+    if image_file is not None:
+        image_path = f"uploads/{name.replace(' ', '_')}_{os.path.basename(image_file)}"
+        os.rename(image_file, image_path)  # Move file to uploads directory
+    else:
+        image_path = None
 
-        return f"Student {name} successfully registered!"
-    except sqlite3.IntegrityError:
-        return "Error: Email already exists. Please use a different email."
-    except Exception as e:
-        return f"An error occurred: {str(e)}"
+    cursor.execute("INSERT INTO students (name, age, image_path) VALUES (?, ?, ?)", (name, age, image_path))
+    conn.commit()
+    conn.close()
+    return f"Student {name} registered successfully!"
 
 # Initialize the database
-init_db()
-
-# Gradio interface
-def main():
-    with gr.Blocks() as demo:
-        gr.Markdown("## Student Registration Form")
-
-        with gr.Row():
-            name = gr.Textbox(label="Name", placeholder="Enter your full name")
-            email = gr.Textbox(label="Email", placeholder="Enter your email address")
-        with gr.Row():
-            phone = gr.Textbox(label="Phone", placeholder="Enter your phone number")
-            course = gr.Textbox(label="Course", placeholder="Enter your course name")
-        image = gr.Image(label="Upload Image", type="file")
-        
-        submit_button = gr.Button("Register")
-        output = gr.Textbox(label="Output")
-        
-        submit_button.click(register_student, inputs=[name, email, phone, course, image], outputs=output)
-
-    return demo
-
-if __name__ == "__main__":
-    app = main()
-    app.launch()
+setup_database()
+
+# Gradio Interface
+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)
+        image_input = gr.Image(label="Upload Image", type="filepath")
+
+    register_button = gr.Button("Register Student")
+    output = gr.Textbox(label="Status")
+
+    register_button.click(
+        register_student,
+        inputs=[name_input, age_input, image_input],
+        outputs=output,
+    )
+
+# Launch the Gradio app
+demo.launch()