Update app.py

app.py

@@ -9,11 +9,33 @@ from embedding_utils import parallel_generate_embeddings, get_embedding
 # Load environment variables from .env file
 load_dotenv()
 
-# Initialize OpenAI client
-openai_client = OpenAI()
+def check_credentials() -> tuple[bool, str]:
+    """Check if required credentials are set and valid"""
+    atlas_uri = os.getenv("ATLAS_URI")
+    openai_key = os.getenv("OPENAI_API_KEY")
+    
+    if not atlas_uri:
+        return False, """Please set up your MongoDB Atlas credentials:
+1. Go to Settings tab
+2. Add ATLAS_URI as a Repository Secret
+3. Paste your MongoDB connection string (should start with 'mongodb+srv://')"""
+    
+    if not openai_key:
+        return False, """Please set up your OpenAI API key:
+1. Go to Settings tab
+2. Add OPENAI_API_KEY as a Repository Secret
+3. Paste your OpenAI API key"""
+        
+    return True, ""
 
-# Initialize database utils
-db_utils = DatabaseUtils()
+def init_clients():
+    """Initialize OpenAI and MongoDB clients"""
+    try:
+        openai_client = OpenAI()
+        db_utils = DatabaseUtils()
+        return openai_client, db_utils
+    except Exception as e:
+        return None, None
 
 def get_field_names(db_name: str, collection_name: str) -> list[str]:
     """Get list of fields in the collection"""
@@ -152,8 +174,29 @@ def vector_search(query_text: str, db_name: str, collection_name: str, embedding
 with gr.Blocks(title="MongoDB Vector Search Tool") as iface:
     gr.Markdown("# MongoDB Vector Search Tool")
     
-    # Get available databases
-    databases = db_utils.get_databases()
+    # Check credentials first
+    has_creds, cred_message = check_credentials()
+    if not has_creds:
+        gr.Markdown(f"""
+        ## ⚠️ Setup Required
+        
+        {cred_message}
+        
+        After setting up credentials, refresh this page.
+        """)
+    else:
+        # Initialize clients
+        openai_client, db_utils = init_clients()
+        if not openai_client or not db_utils:
+            gr.Markdown("""
+            ## ⚠️ Connection Error
+            
+            Failed to connect to MongoDB Atlas or OpenAI. Please check your credentials and try again.
+            """)
+        else:
+            # Get available databases
+            try:
+                databases = db_utils.get_databases()
     
     with gr.Tab("Generate Embeddings"):
         with gr.Row():