Changes to app for env

app.py

@@ -209,23 +209,39 @@ class MCPClient:
 def create_gradio_interface():
     """Create the Gradio interface"""
     
-    # Initialize MCP Client (will be set when API key is provided)
+    # Try to get API key from environment (HuggingFace Secrets)
+    openai_api_key = os.getenv("OPEN_AI_API_KEY")
+    
+    # Initialize MCP Client if API key is available
     mcp_client = None
+    initial_status = ""
+    
+    if openai_api_key:
+        try:
+            mcp_client = MCPClient(openai_api_key)
+            initial_status = "✅ OpenAI client initialized automatically from HF Secrets!"
+        except Exception as e:
+            initial_status = f"❌ Failed to initialize with HF Secret: {str(e)}"
+    else:
+        initial_status = "⚠️ OpenAI API key not found in HF Secrets. Please enter manually."
     
     def initialize_client(api_key):
         nonlocal mcp_client
-        if not api_key:
+        # Use provided key or fall back to environment
+        key_to_use = api_key.strip() if api_key and api_key.strip() else openai_api_key
+        
+        if not key_to_use:
             return "❌ Please provide your OpenAI API key", ""
         
         try:
-            mcp_client = MCPClient(api_key)
+            mcp_client = MCPClient(key_to_use)
             return "✅ OpenAI client initialized successfully!", ""
         except Exception as e:
             return f"❌ Failed to initialize: {str(e)}", ""
     
     def process_message(message, server_choice, history):
         if not mcp_client:
-            return history + [["Please initialize the OpenAI client first.", ""]], ""
+            return history + [["❌ Error", "Please initialize the OpenAI client first using the button above."]], ""
         
         if not message.strip():
             return history, ""
@@ -259,17 +275,23 @@ def create_gradio_interface():
     with gr.Blocks(title="MCP Client with OpenAI", theme=gr.themes.Soft()) as app:
         gr.Markdown("# 🤖 MCP Client with OpenAI Integration")
         gr.Markdown("Connect to your Terraform, Linux, and Cisco MCP servers with AI-powered assistance")
+        gr.Markdown("💡 **Tip**: Your OpenAI API key is automatically loaded from HuggingFace Secrets!")
         
         with gr.Row():
             with gr.Column(scale=3):
-                # API Key input
+                # API Key input (optional if already in HF Secrets)
                 api_key = gr.Textbox(
-                    label="OpenAI API Key",
+                    label="OpenAI API Key (Optional - will use HF Secret if available)",
                     type="password",
-                    placeholder="Enter your OpenAI API key..."
+                    placeholder="Leave empty to use HF Secret OPEN_AI_API_KEY...",
+                    value=""
+                )
+                init_btn = gr.Button("Initialize/Reinitialize Client", variant="primary")
+                init_status = gr.Textbox(
+                    label="Initialization Status", 
+                    interactive=False,
+                    value=initial_status
                 )
-                init_btn = gr.Button("Initialize Client", variant="primary")
-                init_status = gr.Textbox(label="Status", interactive=False)
                 
                 # Server selection
                 server_choice = gr.Dropdown(