Update app.py

app.py

@@ -1,14 +1,18 @@
-# app.py
 import os
 import asyncio
+import logging
 from dotenv import load_dotenv
 import chainlit as cl
 
+# Set up logging
+logging.basicConfig(level=logging.INFO)
+
 # Attempt to import google.generativeai (optional dependency)
 try:
     import google.generativeai as genai
-except Exception:
+except ImportError:
     genai = None
+    logging.warning("google.generativeai not installed or failed to import. Gemini will be unavailable.")
 
 # Load .env (won't override Space secrets)
 load_dotenv()
@@ -19,19 +23,16 @@ MODEL_NAME = os.getenv("MODEL_NAME", "gemini-2.0-flash")
 
 # Try to configure Gemini if possible
 model = None
-if genai is None:
-    print("[Info] google-generativeai not installed or failed to import. Gemini will be unavailable.")
-else:
+if genai is not None:
     if GOOGLE_API_KEY:
         try:
             genai.configure(api_key=GOOGLE_API_KEY)
             model = genai.GenerativeModel(MODEL_NAME)
-            print(f"[Info] Gemini model initialized: {MODEL_NAME}")
+            logging.info(f"Gemini model initialized: {MODEL_NAME}")
         except Exception as e:
-            print(f"[Warning] Failed to init Gemini model '{MODEL_NAME}': {e}")
-            model = None
-    else:
-        print("[Info] GOOGLE_API_KEY not found in environment. Set it in Space Secrets to enable Gemini.")
+            logging.warning(f"Failed to init Gemini model '{MODEL_NAME}': {e}")
+else:
+    logging.info("GOOGLE_API_KEY not found in environment. Set it in Space Secrets to enable Gemini.")
 
 SYSTEM_PROMPT = "You are a helpful assistant."
 
@@ -46,7 +47,6 @@ async def on_chat_start():
     """Welcome message and session init."""
     cl.user_session.set("message_history", [{"role": "system", "content": SYSTEM_PROMPT}])
     if model is None:
-        # Informative welcome when Gemini not available
         await cl.Message(
             content=(
                 "Hello — the app is running. Gemini is not available because GOOGLE_API_KEY is not set or the model failed to init.\n\n"
@@ -64,17 +64,16 @@ async def handle_message(message: cl.Message):
         await cl.Message(content="Please type a question or prompt.").send()
         return
 
-    # Store message in session history (for bookkeeping)
+    # Store message in session history
     history = cl.user_session.get("message_history", [])
     history.append({"role": "user", "content": user_text})
     cl.user_session.set("message_history", history)
 
-    # Send a progress message that we'll update
+    # Send a progress message
     progress = cl.Message(content="(Thinking...)")
     await progress.send()
 
     if model is None:
-        # Fallback behavior: echo plus short helpful tips
         reply = (
             "Gemini not initialized (GOOGLE_API_KEY missing or model failed to init).\n\n"
             "Echoing your input:\n\n" + user_text + "\n\n"
@@ -84,16 +83,17 @@ async def handle_message(message: cl.Message):
         try:
             resp = await _call_gemini_generate(user_text)
             reply = getattr(resp, "text", None) or str(resp)
+        except RuntimeError as e:
+            reply = f"[Runtime Error] {e}"
         except Exception as e:
             reply = f"[Error calling Gemini API] {e}"
 
     # Update the progress message with final reply
     try:
-        progress.content = reply
-        await progress.update()
-    except Exception:
+        await progress.update(content=reply)
+    except Exception as e:
         await cl.Message(content=reply).send()
 
     # Save assistant reply to history
     history.append({"role": "assistant", "content": reply})
-    cl.user_session.set("message_history", history)
+    cl.user_session.set("message_history", history)