Update app.py

app.py

@@ -43,11 +43,10 @@ DATABASE_URI = f"sqlite:///{os.path.join(BASE_DIR, 'data', 'mydb.db')}"
 print("DATABASE URI:", DATABASE_URI)
 
 # API Keys from .env file
-import os
 os.environ["GROQ_API_KEY"] = os.getenv("GROQ_API_KEY")
 os.environ["MISTRAL_API_KEY"] = os.getenv("MISTRAL_API_KEY")
 
-# Global variables for dynamic agent and DB file path; initially None.
+# Global state: dynamic agent and DB file path
 agent_app = None
 abs_file_path = None
 db_path = None
@@ -56,7 +55,7 @@ db_path = None
 # create_agent_app: Given a database path, initialize the agent workflow.
 # =============================================================================
 def create_agent_app(db_path: str):
-    # Use ChatGroq as our LLM here; you can swap to ChatMistralAI if preferred.
+    # Use ChatGroq as our LLM here; swap to ChatMistralAI if preferred.
     from langchain_groq import ChatGroq
     llm = ChatGroq(model="llama3-70b-8192")
 
@@ -87,7 +86,7 @@ def create_agent_app(db_path: str):
         messages: Annotated[list[AnyMessage], add_messages]
 
     # -------------------------------------------------------------------------
-    # Set up prompt templates for query checking and query generation.
+    # Set up prompt templates for query checking and generation.
     # -------------------------------------------------------------------------
     query_check_system = (
         "You are a SQL expert with a strong attention to detail.\n"
@@ -245,7 +244,6 @@ def create_app():
         if not abs_file_path:
             socketio.emit("log", {"message": "[ERROR]: No DB file uploaded."})
             socketio.emit("final", {"message": "No database available. Please upload one and try again."})
-            flash("No database available. Please upload one and try again.", "error")
             return
         try:
             if agent_app is None:
@@ -306,7 +304,7 @@ def create_app():
                     print("Saving file to:", db_path)
                     file.save(db_path)
                     abs_file_path = os.path.abspath(db_path)
-                    agent_app = None   # Reset the agent so it is lazily reinitialized on next query.
+                    agent_app = None   # Reset the agent to be lazily reinitialized on next query.
                     print(f"[INFO]: File '{filename}' uploaded. Agent will be initialized on first query.")
                     socketio.emit("log", {"message": f"[INFO]: Database file '{filename}' uploaded."})
                     flash(f"Database file '{filename}' uploaded successfully.", "info")