chart code with DB

main.py

@@ -27,7 +27,11 @@ async def show_chart(data: List[dict]):
     categories = [item["category"] for item in data]
     values = [item["value"] for item in data]
 
-    return HTMLResponse(content=create_chart(categories, values), status_code=200)
+    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)
 
 @app.get("/visualization/", response_class=HTMLResponse)
 async def show_visualization():
@@ -36,4 +40,8 @@ async def show_visualization():
     categories = ["Category1", "Category2", "Category3"]
     values = [10, 20, 30]
 
-    return HTMLResponse(content=create_chart(categories, values), status_code=200)
+    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)