Update main.py

main.py

@@ -1,5 +1,5 @@
 from fastapi import FastAPI, HTTPException
-from pydantic import BaseModel
+from pydantic import BaseModel, EmailStr
 import random
 import string
 from datetime import datetime
@@ -19,6 +19,7 @@ if not DB_FILE.exists():
 # Data model
 class Data(BaseModel):
     number: int
+    email: EmailStr  # Add the email field with validation
 
 # Generate unique ID
 def generate_id():
@@ -42,6 +43,7 @@ def add_data(data: Data):
         "id": generate_id(),
         "date": datetime.utcnow().isoformat(),
         "number": data.number,
+        "email": data.email,  # Include the email field
     }
     database.append(new_entry)
     write_database(database)
@@ -55,10 +57,7 @@ def get_data():
 @app.get("/data/{item_id}")
 def get_data_by_id(item_id: str):
     database = read_database()
-    print(f"Database: {database}")  # Debugging: print the database
     for entry in database:
-        print(f"Checking entry: {entry}")  # Debugging: print each entry
         if entry["id"] == item_id:
             return entry
     raise HTTPException(status_code=404, detail="Item not found")
-