cleanup

app.py

@@ -1,14 +1,11 @@
-from os import write
-from langchain.llms.base import LLM
 import streamlit as st
 import numpy as np
 import pandas as pd
 import altair as alt
 import chat as idf_chat
-# import sqlite3
 from langchain.sql_database import SQLDatabase
 from langchain.agents.agent_toolkits import SQLDatabaseToolkit
-from langchain.agents import LLMSingleActionAgent, agent_types, create_sql_agent, initialize_agent, AgentType,load_tools
+from langchain.agents import create_sql_agent
 from langchain import OpenAI
 from langchain import PromptTemplate, OpenAI, LLMChain
 from langchain.chains import SimpleSequentialChain
@@ -27,9 +24,6 @@ if DB_CHAIN not in st.session_state:
     st.session_state[DB_CHAIN] = {}
 
 def get_sql_agent():
-#    db = SQLDatabase.from_uri("sqlite:///switrs.sqlite")
-    db = SQLDatabase.from_uri("sqlite:///FXTrades.db")
-
     toolkit = SQLDatabaseToolkit(llm=llm,db=db)
 
     return create_sql_agent(