chart code with DB

main.py

@@ -1,24 +1,47 @@
 from fastapi import FastAPI, HTTPException, Body
-import pandas as pd
+from sqlalchemy import create_engine, Column, Integer, String
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+from fastapi.responses import HTMLResponse
+import gradio as gr
 
 app = FastAPI()
 
-def process_json(input_data: dict):
+SQLALCHEMY_DATABASE_URL = "sqlite:///./test.db"
+
+engine = create_engine(SQLALCHEMY_DATABASE_URL)
+SessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=engine)
+Base = declarative_base()
+
+def create_gr_chart(data):
+    chart = gr.LineChart(data=data)
+    return chart.launch()
+class Item(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:
-        # Assuming the input_data is a dictionary with keys representing columns and values as lists
-        df = pd.DataFrame(input_data)
-    except Exception as e:
-        raise HTTPException(status_code=400, detail=f"Error processing input data: {str(e)}")
-
-    # Customize this part based on your specific charting requirements
-    # For example, you can return a dictionary with keys for x-axis and y-axis data
-    chart_data = {
-        "x_axis": df.columns.tolist(),
-        "y_axis": df.values.tolist()
-    }
-
-    return chart_data
-
-@app.post("/create-chart")
-async def create_chart(input_data: dict = Body(...)):
-    return process_json(input_data)
+        yield db
+    finally:
+        db.close()
+
+@app.post("/form/")
+async def store_data(item: Item, db: Session = Depends(get_db)):
+    db.add(item)
+    db.commit()
+    db.refresh(item)
+    return item
+
+@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()  # Define this function to fetch data from the database
+    chart = create_gr_chart(data)  # Define this function to create a Gradio chart
+    return chart

requirements.txt

@@ -1,3 +1,5 @@
 fastapi
 uvicorn
-pandas
+pandas
+sqlite
+gradio