Update app.py

app.py

@@ -407,6 +407,17 @@ def load_text_examples():
     
     return examples
 
+# Function to check API availability - modified to work directly
+def check_api_status():
+    # Check Gemini API availability
+    api_key = os.getenv("GAPI_TOKEN")
+    gemini_available = api_key is not None
+    
+    if gemini_available:
+        return """<div class="api-status api-connected">✓ Connected to FLUX.1 and Gemini API</div>"""
+    else:
+        return """<div class="api-status api-disconnected">✗ Gemini API connection issue. Please check GAPI_TOKEN environment variable.</div>"""
+
 # CSS for improved UI
 css = """
 :root {
@@ -588,11 +599,8 @@ with gr.Blocks(css=css) as demo:
     </div>
     """)
     
-    # API Status
-    api_status = gr.Markdown(
-        """<div class="api-status api-connected">✓ Connected to image generation and Gemini API</div>""", 
-        visible=True
-    )
+    # API Status - 직접 호출해서 초기 상태 설정
+    api_status = gr.Markdown(check_api_status(), visible=True)
     
     with gr.Tabs():
         with gr.Tab("Create Ghibli Art"):
@@ -774,19 +782,5 @@ with gr.Blocks(css=css) as demo:
     </div>
     """)
 
-# Function to check API availability
-def check_api_status():
-    # Check Gemini API availability
-    api_key = os.getenv("GAPI_TOKEN")
-    gemini_available = api_key is not None
-    
-    if gemini_available:
-        return """<div class="api-status api-connected">✓ Connected to FLUX.1 and Gemini API</div>"""
-    else:
-        return """<div class="api-status api-disconnected">✗ Gemini API connection issue. Please check GAPI_TOKEN environment variable.</div>"""
-
-# Update API status on page load
-demo.load(fn=check_api_status, outputs=api_status)
-
 # Launch the Gradio app
 demo.queue().launch()