chart code with DB

main.py

@@ -1,58 +1,32 @@
-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
+from fastapi import FastAPI, HTTPException, Form
 from fastapi.responses import HTMLResponse
 import gradio as gr
-import os
-import logging
-from fastapi import Depends
 
 app = FastAPI()
 
-# Use SQLite in-memory database
-SQLALCHEMY_DATABASE_URL = "sqlite:///:memory:"
-
-engine = create_engine(SQLALCHEMY_DATABASE_URL)
-SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
-Base = declarative_base()
-
-class Item(BaseModel):
-    category: str
-    value: int
-
-class ItemDB(Base):
-    __tablename__ = "items"
-
-    id = Column(Integer, primary_key=True, index=True)
-    category = Column(String, index=True)
-    value = Column(Integer)
-
-Base.metadata.create_all(bind=engine)
-
-def get_db():
-    db = SessionLocal()
-    try:
-        yield db
-    finally:
-        db.close()
-
-def create_gr_chart(data):
+def create_gr_chart(categories, values):
+    data = {"x": categories, "y": values}
     chart = gr.LineChart(data=data)
     return chart.launch()
 
 @app.post("/form/")
-async def store_data(item: Item, db: Session = Depends(get_db)):
-    db_item = ItemDB(**item.dict())
-    db.add(db_item)
-    db.commit()
-    db.refresh(db_item)
-    return db_item
+async def show_chart(category: str = Form(...), value: int = Form(...)):
+    # You can append the received category and value to a list if needed
+    # categories.append(category)
+    # values.append(value)
+    
+    # For simplicity, I'm assuming you have two lists of categories and values
+    # Modify this part based on your actual data structure
+    categories = ["Category1", "Category2", "Category3"]
+    values = [10, 20, 30]
+
+    return create_gr_chart(categories, values)
 
 @app.get("/visualization/", response_class=HTMLResponse)
 async def show_visualization():
-    # Fetch data from the database and create a Gradio chart
-    data = get_data_from_db()  # Replace with your function to fetch data from the database
-    chart = create_gr_chart(data)  # Replace with your function to create a Gradio chart
-    return chart
+    # For simplicity, I'm assuming you have two lists of categories and values
+    # Modify this part based on your actual data structure
+    categories = ["Category1", "Category2", "Category3"]
+    values = [10, 20, 30]
+
+    return create_gr_chart(categories, values)