Update app.py

app.py

@@ -1,59 +1,26 @@
+# app.py
+
 from fastapi import FastAPI, HTTPException
+from pydantic import BaseModel # <--- Import BaseModel
 import sqlite3
 import os
 
-app = FastAPI()
-
-# Define the path for your SQLite database file
-# This will be created in the /app directory of your Hugging Face Space.
-# Files in /app are generally persisted across restarts within a Space.
-DATABASE_FILE = os.path.join(os.getcwd(), "data.db")
-
-def get_db_connection():
-    try:
-        conn = sqlite3.connect(DATABASE_FILE)
-        # Configure connection to return rows as dictionaries for easier access
-        conn.row_factory = sqlite3.Row
-        return conn
-    except sqlite3.Error as err:
-        print(f"Error connecting to database: {err}")
-        raise HTTPException(status_code=500, detail="Database connection error")
+# Define the Pydantic model for Item creation/update
+class Item(BaseModel):
+    name: str
+    description: str | None = None # Use Union type hint for None
 
-@app.on_event("startup")
-async def startup_event():
-    # Create tables if they don't exist
-    conn = None # Initialize conn to None
-    try:
-        conn = get_db_connection()
-        cursor = conn.cursor()
-        cursor.execute("""
-            CREATE TABLE IF NOT EXISTS items (
-                id INTEGER PRIMARY KEY AUTOINCREMENT,
-                name TEXT NOT NULL,
-                description TEXT
-            )
-        """)
-        conn.commit()
-        print(f"Database table 'items' checked/created successfully at {DATABASE_FILE}.")
-    except Exception as e:
-        print(f"Error during database startup: {e}")
-        # In a real app, you might want to handle this more gracefully,
-        # e.g., log the error and potentially shut down if essential tables can't be created.
-    finally:
-        if conn:
-            conn.close()
+app = FastAPI()
 
-@app.get("/")
-async def root():
-    return {"message": "Welcome to the API!"}
+# ... rest of your existing code ...
 
 @app.post("/items/")
-async def create_item(name: str, description: str = None):
+async def create_item(item: Item): # <--- Change parameter to 'item: Item'
     conn = get_db_connection()
     cursor = conn.cursor()
     try:
         query = "INSERT INTO items (name, description) VALUES (?, ?)"
-        cursor.execute(query, (name, description))
+        cursor.execute(query, (item.name, item.description)) # Access attributes using item.name, item.description
         conn.commit()
         return {"message": "Item created successfully", "id": cursor.lastrowid}
     except Exception as e:
@@ -63,50 +30,21 @@ async def create_item(name: str, description: str = None):
         cursor.close()
         conn.close()
 
-@app.get("/items/")
-async def read_items():
-    conn = get_db_connection()
-    cursor = conn.cursor()
-    try:
-        cursor.execute("SELECT * FROM items")
-        items = cursor.fetchall()
-        # Convert sqlite3.Row objects to dictionaries for JSON serialization
-        return {"items": [dict(item) for item in items]}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error reading items: {e}")
-    finally:
-        cursor.close()
-        conn.close()
-
-@app.get("/items/{item_id}")
-async def read_item(item_id: int):
-    conn = get_db_connection()
-    cursor = conn.cursor()
-    try:
-        cursor.execute("SELECT * FROM items WHERE id = ?", (item_id,))
-        item = cursor.fetchone()
-        if item is None:
-            raise HTTPException(status_code=404, detail="Item not found")
-        return dict(item)
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=f"Error reading item: {e}")
-    finally:
-        cursor.close()
-        conn.close()
-
+# ... Also update your @app.put("/items/{item_id}") endpoint similarly ...
 @app.put("/items/{item_id}")
-async def update_item(item_id: int, name: str = None, description: str = None):
+async def update_item(item_id: int, item: Item): # <--- Change parameter to 'item: Item'
     conn = get_db_connection()
     cursor = conn.cursor()
     try:
         updates = []
         params = []
-        if name is not None:
+
+        if item.name is not None: # Access using item.name
             updates.append("name = ?")
-            params.append(name)
-        if description is not None:
+            params.append(item.name)
+        if item.description is not None: # Access using item.description
             updates.append("description = ?")
-            params.append(description)
+            params.append(item.description)
 
         if not updates:
             raise HTTPException(status_code=400, detail="No fields to update provided")
@@ -129,19 +67,4 @@ async def update_item(item_id: int, name: str = None, description: str = None):
         cursor.close()
         conn.close()
 
-@app.delete("/items/{item_id}")
-async def delete_item(item_id: int):
-    conn = get_db_connection()
-    cursor = conn.cursor()
-    try:
-        cursor.execute("DELETE FROM items WHERE id = ?", (item_id,))
-        conn.commit()
-        if cursor.rowcount == 0:
-            raise HTTPException(status_code=404, detail="Item not found")
-        return {"message": "Item deleted successfully"}
-    except Exception as e:
-        conn.rollback()
-        raise HTTPException(status_code=500, detail=f"Error deleting item: {e}")
-    finally:
-        cursor.close()
-        conn.close()
+# ... rest of your existing code ...