Update app.py

app.py

@@ -9,12 +9,17 @@ from io import BytesIO
 import tempfile
 
 # --- Google Gemini API Configuration ---
-GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY", "")
-if not GOOGLE_API_KEY:
-    raise ValueError("GOOGLE_API_KEY environment variable not set.")
+# Use GEMINI_API_KEY if available, otherwise fall back to GOOGLE_API_KEY
+GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
+GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
+
+API_KEY = GEMINI_API_KEY or GOOGLE_API_KEY
+
+if not API_KEY:
+    raise ValueError("Neither GEMINI_API_KEY nor GOOGLE_API_KEY environment variable is set.")
 
 client = genai.Client(
-    api_key=os.environ.get("GEMINI_API_KEY"),
+    api_key=API_KEY,
 )
 
 GEMINI_MODEL_NAME = 'gemini-2.5-flash-image-preview'
@@ -237,14 +242,14 @@ button.selected {
 }
 
 /* Examples Styling */
-.examples-container {
+.examples {
     background: #f9fafb;
     border-radius: 0.5rem;
     padding: 1rem;
     margin-top: 1rem;
 }
 
-.dark .examples-container {
+.dark .examples {
     background: #1f2937;
 }
 
@@ -377,8 +382,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as demo:
                             ["봄날 벚꽃이 만개한 일본 정원"],
                             ["판타지 세계의 마법사 탑"],
                         ],
-                        inputs=prompt_input,
-                        elem_classes="examples-container"
+                        inputs=prompt_input
                     )
                 
                 gr.HTML('</div>')