Update main.py

main.py

@@ -63,10 +63,14 @@ def create_agent(filename: str):
     llm = ChatOpenAI(temperature=0, model="gpt-4o-2024-05-13")
     #llm = OpenAI(temperature=0, model_name="gpt-4")
     # Read the CSV file into a Pandas DataFrame.
-    df = pd.read_csv(filename)
-    engine = create_engine("sqlite:///sphinx.db")
-    df.to_sql("sphinx", engine, index=False)
-    db = SQLDatabase(engine=engine)
+    if db.get_usable_table_names() == None:
+        df = pd.read_csv(filename)
+        engine = create_engine("sqlite:///sphinx.db")
+        df.to_sql("sphinx", engine, index=False)
+        db = SQLDatabase(engine=engine)
+        cl.user_session.set("db", db)
+    else:
+        db = cl.user_session.get("db")
     # Create a SAL agent.
     #e.g agent_executor.invoke({"input": "Quel est le nombre de chargé d'affaires en agencement par entreprise?"})
     #return create_pandas_dataframe_agent(llm, df, verbose=False, allow_dangerous_code=True, handle_parsing_errors=True, agent_type=AgentType.OPENAI_FUNCTIONS)