chart code with DB

main.py

@@ -1,37 +1,18 @@
 from fastapi import FastAPI, HTTPException
 from fastapi.responses import HTMLResponse
-from typing import List
-import matplotlib.pyplot as plt
-import io
-import base64
+import gradio as gr
 
 app = FastAPI()
 
 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}"/>'
+    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
 
 @app.post("/form/")
-async def show_chart(data: List[dict]):
-    categories = [item["category"] for item in data]
-    values = [item["value"] for item in data]
-
-    chart_html = create_chart(categories, values)
-
-    # Embed the chart HTML in an iframe
-    iframe_html = f'<iframe width="600" height="400" srcdoc="{chart_html}"></iframe>'
-    return HTMLResponse(content=iframe_html, status_code=200)
+async def show_chart(categories: list, values: list):
+    chart = create_chart(categories, values)
+    return HTMLResponse(content=chart.launch(), status_code=200)
 
 @app.get("/visualization/", response_class=HTMLResponse)
 async def show_visualization():
@@ -40,8 +21,5 @@ async def show_visualization():
     categories = ["Category1", "Category2", "Category3"]
     values = [10, 20, 30]
 
-    chart_html = create_chart(categories, values)
-
-    # Embed the chart HTML in an iframe
-    iframe_html = f'<iframe width="600" height="400" srcdoc="{chart_html}"></iframe>'
-    return HTMLResponse(content=iframe_html, status_code=200)
+    chart = create_chart(categories, values)
+    return HTMLResponse(content=chart.launch(), status_code=200)