chart code with DB

main.py

@@ -1,21 +1,33 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import HTMLResponse
 from typing import List
-import gradio as gr
+import matplotlib.pyplot as plt
+import io
+import base64
 
 app = FastAPI()
 
-def create_gr_chart(categories, values):
-    data = {"x": categories, "y": values}
-    chart = gr.LineChart(data=data)
-    return chart.launch()
+def create_chart(categories, values):
+    plt.plot(categories, values)
+    plt.xlabel('Categories')
+    plt.ylabel('Values')
+    
+    # Save the plot to a BytesIO object
+    img = io.BytesIO()
+    plt.savefig(img, format='png')
+    img.seek(0)
+    plt.close()
+
+    # Convert the BytesIO object to base64 for embedding in HTML
+    img_base64 = base64.b64encode(img.read()).decode("utf-8")
+    return f'<img src="data:image/png;base64,{img_base64}"/>'
 
 @app.post("/form/")
 async def show_chart(data: List[dict]):
     categories = [item["category"] for item in data]
     values = [item["value"] for item in data]
 
-    return create_gr_chart(categories, values)
+    return HTMLResponse(content=create_chart(categories, values), status_code=200)
 
 @app.get("/visualization/", response_class=HTMLResponse)
 async def show_visualization():
@@ -24,4 +36,4 @@ async def show_visualization():
     categories = ["Category1", "Category2", "Category3"]
     values = [10, 20, 30]
 
-    return create_gr_chart(categories, values)
+    return HTMLResponse(content=create_chart(categories, values), status_code=200)