Update app.py

app.py

@@ -1,9 +1,8 @@
 import os
 import time
-import json
 import pandas as pd
-import numpy as np
 from sqlalchemy import create_engine
+from sqlalchemy.types import DateTime, Float, Integer
 from langchain_openai import ChatOpenAI
 from langchain_community.agent_toolkits import create_sql_agent
 from langchain_community.utilities import SQLDatabase
@@ -14,14 +13,13 @@ import logging
 
 load_dotenv()
 
-JSON_FILE_PATH = "model_training_part1.json"
+CSV_FILE_PATH = "model_training_part1.json"
 SQL_DB_PATH = "data.db"
-HF_API_KEY = os.getenv("HF_API_KEY")
+HUGGINGFACE_API_KEY = os.getenv("HUGGINGFACE_API_KEY")
 OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 LLAMA_MODEL = "meta-llama/Llama-3.3-70B-Instruct"
-LLAMA_MODEL_ALTERNATIVO = "meta-llama/Llama-3.1-8B-Instruct"
 
-hf_client = InferenceClient(api_key=HF_API_KEY)
+hf_client = InferenceClient(api_key=HUGGINGFACE_API_KEY)
 os.environ["OPENAI_API_KEY"] = OPENAI_API_KEY
 
 query_cache = {}
@@ -32,7 +30,7 @@ advanced_mode_enabled = False
 
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
 
-def create_or_load_sql_database(json_path, sql_db_path):
+def create_or_load_sql_database(csv_path, sql_db_path):
     if os.path.exists(sql_db_path):
         print("Banco de dados SQL já existe. Carregando...")
         return create_engine(f"sqlite:///{sql_db_path}")
@@ -40,7 +38,7 @@ def create_or_load_sql_database(json_path, sql_db_path):
         print("Banco de dados SQL não encontrado. Criando...")
         engine = create_engine(f"sqlite:///{sql_db_path}")
 
-        with open(json_path, "r", encoding="utf-8") as f:
+        with open(csv_path, "r", encoding="utf-8") as f:
             data = json.load(f)
 
         df = pd.DataFrame(data)
@@ -48,7 +46,7 @@ def create_or_load_sql_database(json_path, sql_db_path):
         print("Banco de dados SQL criado com sucesso!")
         return engine
 
-engine = create_or_load_sql_database(JSON_FILE_PATH, SQL_DB_PATH)
+engine = create_or_load_sql_database(CSV_FILE_PATH, SQL_DB_PATH)
 db = SQLDatabase(engine=engine)
 
 llm = ChatOpenAI(model="gpt-4o-mini", temperature=0.2)