debuging

app.py

@@ -48,11 +48,31 @@ def setup_database_connection():
     try:
         load_dotenv(override=True)
         
+        # Debug: Log all environment variables (without sensitive values)
+        logger.info("Environment variables:")
+        for key, value in os.environ.items():
+            if any(s in key.lower() for s in ['pass', 'key', 'secret']):
+                logger.info(f"  {key}: {'*' * 8} (hidden for security)")
+            else:
+                logger.info(f"  {key}: {value}")
+        
         db_user = os.getenv("DB_USER")
         db_password = os.getenv("DB_PASSWORD")
         db_host = os.getenv("DB_HOST")
         db_name = os.getenv("DB_NAME")
         
+        # Debug: Log database connection info (without password)
+        logger.info(f"Database connection attempt - Host: {db_host}, User: {db_user}, DB: {db_name}")
+        if not all([db_user, db_password, db_host, db_name]):
+            missing = [var for var, val in [
+                ("DB_USER", db_user),
+                ("DB_PASSWORD", "*" if db_password else ""),
+                ("DB_HOST", db_host),
+                ("DB_NAME", db_name)
+            ] if not val]
+            logger.error(f"Missing required database configuration: {', '.join(missing)}")
+            return None, f"Missing database configuration: {', '.join(missing)}"
+        
         if not all([db_user, db_password, db_host, db_name]):
             return None, "Missing database configuration"
             
@@ -82,37 +102,74 @@ def setup_database_connection():
 def initialize_llm():
     """Inicializa el modelo de lenguaje."""
     if not DEPENDENCIES_AVAILABLE:
-        return None, "Dependencies not available"
+        error_msg = "Dependencies not available. Make sure all required packages are installed."
+        logger.error(error_msg)
+        return None, error_msg
         
     google_api_key = os.getenv("GOOGLE_API_KEY")
+    logger.info(f"GOOGLE_API_KEY found: {'Yes' if google_api_key else 'No'}")
+    
     if not google_api_key:
-        return None, "GOOGLE_API_KEY not found in environment variables"
+        error_msg = "GOOGLE_API_KEY not found in environment variables. Please check your Hugging Face Space secrets."
+        logger.error(error_msg)
+        return None, error_msg
         
     try:
+        logger.info("Initializing Google Generative AI...")
         llm = ChatGoogleGenerativeAI(
             model="gemini-2.0-flash",
             temperature=0,
             google_api_key=google_api_key
         )
+        
+        # Test the model with a simple prompt
+        test_prompt = "Hello, this is a test."
+        logger.info(f"Testing model with prompt: {test_prompt}")
+        test_response = llm.invoke(test_prompt)
+        logger.info(f"Model test response: {str(test_response)[:100]}...")  # Log first 100 chars
+        
         logger.info("Google Generative AI initialized successfully")
         return llm, ""
+        
     except Exception as e:
         error_msg = f"Error initializing Google Generative AI: {str(e)}"
-        logger.error(error_msg)
+        logger.error(error_msg, exc_info=True)  # Include full stack trace
         return None, error_msg
 
 def create_agent():
     """Crea el agente SQL si es posible."""
     if not DEPENDENCIES_AVAILABLE:
-        return None, "Dependencies not available"
-        
+        error_msg = "Dependencies not available. Please check if all required packages are installed."
+        logger.error(error_msg)
+        return None, error_msg
+    
+    logger.info("Starting agent creation process...")
+    
+    # Step 1: Set up database connection
+    logger.info("Setting up database connection...")
     db, db_error = setup_database_connection()
+    if not db:
+        error_msg = f"Failed to connect to database: {db_error}"
+        logger.error(error_msg)
+    else:
+        logger.info("Database connection successful")
+    
+    # Step 2: Initialize LLM
+    logger.info("Initializing language model...")
     llm, llm_error = initialize_llm()
+    if not llm:
+        error_msg = f"Failed to initialize language model: {llm_error}"
+        logger.error(error_msg)
+    else:
+        logger.info("Language model initialized successfully")
     
+    # Check if both components are available
     if not db or not llm:
-        error_msg = " | ".join(filter(None, [db_error, llm_error]))
-        return None, f"Cannot create agent: {error_msg}"
+        error_msg = f"Cannot create agent. {db_error if not db else ''} {llm_error if not llm else ''}"
+        logger.error(error_msg)
+        return None, error_msg
     
+    # Step 3: Create SQL agent
     try:
         logger.info("Creating SQL agent...")
         agent = create_sql_agent(
@@ -121,11 +178,21 @@ def create_agent():
             agent_type=AgentType.OPENAI_FUNCTIONS,
             verbose=True
         )
-        logger.info("SQL agent created successfully")
+        
+        # Test the agent with a simple query
+        try:
+            logger.info("Testing agent with a simple query...")
+            test_result = agent.invoke({"input": "What tables are available?"})
+            logger.info(f"Agent test response: {str(test_result)[:200]}...")  # Log first 200 chars
+        except Exception as test_error:
+            logger.warning(f"Agent test query failed (this might be expected): {str(test_error)}")
+        
+        logger.info("SQL agent created and tested successfully")
         return agent, ""
+        
     except Exception as e:
         error_msg = f"Error creating SQL agent: {str(e)}"
-        logger.error(error_msg)
+        logger.error(error_msg, exc_info=True)  # Include full stack trace
         return None, error_msg
 
 # Inicializar el agente