chart code with DB

main.py

@@ -1,9 +1,8 @@
-from fastapi import FastAPI, HTTPException, Body, Depends
+from fastapi import FastAPI, Depends
 from sqlalchemy import create_engine, Column, Integer, String
 from sqlalchemy.ext.declarative import declarative_base
-from sqlalchemy.orm import sessionmaker
+from sqlalchemy.orm import sessionmaker, Session  # Add this import
 from fastapi.responses import HTMLResponse
-from fastapi import Depends
 import gradio as gr
 import os
 import logging