chart code with DB

main.py

@@ -5,14 +5,15 @@ import gradio as gr
 app = FastAPI()
 
 def create_chart(categories, values):
-    data = {"students": categories, "marks": values}
-    plot = gr.LinePlot(data, every=5, x="students", y="marks", y_title="marks (percent)", overlay_point=True, width=500, height=500)
-    return plot
+    data = {"Date": categories, "Sales": values}
+    plot = gr.LinePlot(data, every=5, x="Date", y="Sales", y_title="Sales ($ millions)", overlay_point=True, width=500, height=500)
+    return plot.render()
 
 @app.post("/form/")
 async def show_chart(categories: list, values: list):
-    chart = create_chart(categories, values)
-    return HTMLResponse(content=chart.launch(), status_code=200)
+    chart_html = create_chart(categories, values)
+    iframe_code = f'<iframe srcdoc="{chart_html}" width="600" height="400" frameborder="0"></iframe>'
+    return HTMLResponse(content=iframe_code, status_code=200)
 
 @app.get("/visualization/", response_class=HTMLResponse)
 async def show_visualization():
@@ -21,5 +22,6 @@ async def show_visualization():
     categories = ["Category1", "Category2", "Category3"]
     values = [10, 20, 30]
 
-    chart = create_chart(categories, values)
-    return HTMLResponse(content=chart.launch(), status_code=200)
+    chart_html = create_chart(categories, values)
+    iframe_code = f'<iframe srcdoc="{chart_html}" width="600" height="400" frameborder="0"></iframe>'
+    return HTMLResponse(content=iframe_code, status_code=200)