app.py ui improvements

app.py

@@ -55,13 +55,18 @@ def encode_image(data: np.ndarray) -> dict:
         base64_str = str(base64.b64encode(bytes_data), "utf-8")
         return {"mime_type": "image/jpeg", "data": base64_str}
     except Exception as e:
+        # log traceback
+        logger.error(traceback.format_exc())
         logger.error(f"Error encoding image: {e}")
         raise
+    except ValueError as e: # PIL error
+        logger.exception(f"ValueError when creating image: {e}")
+        raise
 
 def check_twilio_availability_sync() -> bool:
     """Checks Twilio TURN server availability (synchronous version)."""
     global twilio_available
-    timeout = 10
+    timeout_seconds = 10
     retries = 3
     delay = 2
 
@@ -233,9 +238,9 @@ class GeminiHandler(AsyncAudioVideoStreamHandler):
 
 def update_gemini_status_sync():
     """Updates the Gemini status message (synchronous version)."""
-    status = "Connected" if gemini_connected else "Disconnected"
+    status = "✅ Gemini: Connected" if gemini_connected else "❌ Gemini: Disconnected"
     if 'demo' in locals() and demo.running:
-        gr.update(value=f"Gemini connection status: {status}")
+        gr.update(value=status)
 
 
 
@@ -265,8 +270,8 @@ with gr.Blocks(css=css) as demo:
     </div>
     """
     )
-    twilio_status_message = gr.Markdown("")
-    gemini_status_message = gr.Markdown("")
+    twilio_status_message = gr.Markdown("❓ Twilio: Checking...")
+    gemini_status_message = gr.Markdown("❓ Gemini: Checking...")
 
     with gr.Row() as api_key_row:
         api_key = gr.Textbox(
@@ -298,15 +303,16 @@ with gr.Blocks(css=css) as demo:
 
     def update_twilio_status_ui():
         if twilio_available:
-            message = "Twilio TURN server available."
+            message = "✅ Twilio: Available"
         else:
-            message = "**Warning:** Twilio TURN server unavailable. Connection might be less reliable."
+            message = "❌ Twilio: Unavailable (connection may be less reliable)"
         return gr.update(value=message)
 
     demo.load(update_twilio_status_ui, [], [twilio_status_message])
 
+    handler = GeminiHandler()
     webrtc.stream(
-        GeminiHandler(),
+        handler,
         inputs=[webrtc, api_key, image_input],
         outputs=[webrtc],
         time_limit=90,
@@ -320,13 +326,13 @@ with gr.Blocks(css=css) as demo:
                 gr.update(visible=True),
                 gr.update(visible=False),
                 gr.update(value="Please enter a valid API key"),
-                gr.update(value=""),
+                gr.update(value="❓ Gemini: Checking..."),
             )
         return (
             gr.update(visible=False),
             gr.update(visible=True),
             gr.update(value=""),
-            gr.update(value=""),
+            gr.update(value="❓ Gemini: Checking..."),
         )
 
     api_key.submit(
@@ -335,4 +341,12 @@ with gr.Blocks(css=css) as demo:
         [api_key_row, row, twilio_status_message, gemini_status_message],
     )
 
+    # If API key is already set via environment variables, hide the API key row and show content
+    if os.getenv("GOOGLE_API_KEY"):
+        demo.load(
+            lambda: (gr.update(visible=False), gr.update(visible=True)),
+            None,
+            [api_key_row, row],
+        )
+
 demo.launch()