chart code with DB

main.py

@@ -1,7 +1,8 @@
+from pydantic import BaseModel
 from fastapi import FastAPI, Depends
 from sqlalchemy import create_engine, Column, Integer, String
 from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import sessionmaker, Session  # Add this import
+from sqlalchemy.orm import sessionmaker, Session
 from fastapi.responses import HTMLResponse
 import gradio as gr
 import os
@@ -16,7 +17,11 @@ engine = create_engine(SQLALCHEMY_DATABASE_URL)
 SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
 Base = declarative_base()
 
-class Item(Base):
+class Item(BaseModel):
+    category: str
+    value: int
+
+class ItemDB(Base):
     __tablename__ = "items"
 
     id = Column(Integer, primary_key=True, index=True)
@@ -38,10 +43,11 @@ def create_gr_chart(data):
 
 @app.post("/form/")
 async def store_data(item: Item, db: Session = Depends(get_db)):
-    db.add(item)
+    db_item = ItemDB(**item.dict())
+    db.add(db_item)
     db.commit()
-    db.refresh(item)
-    return item
+    db.refresh(db_item)
+    return db_item
 
 @app.get("/visualization/", response_class=HTMLResponse)
 async def show_visualization():