Update streamlit_app.py

streamlit_app.py

@@ -21,11 +21,14 @@ try:
     from phi.tools.duckduckgo import DuckDuckGo
     HAS_PHI = True
 except Exception:
-# disable phi Agent usage to avoid phi IndexError in production
+    Agent = Gemini = DuckDuckGo = None
     HAS_PHI = False
-    Agent = None
-    Gemini = None
-    DuckDuckGo = None
+
+# disable phi Agent to avoid phi IndexError in production by default
+HAS_PHI = False
+Agent = None
+Gemini = None
+DuckDuckGo = None
 
 try:
     import google.generativeai as genai
@@ -347,7 +350,7 @@ if generate_now and not st.session_state.get("busy"):
 
                 out = ""
                 model_used = model_id
-                max_tokens = 512 if st.session_state.get("fast_mode") else 1024
+                max_tokens = 256 if st.session_state.get("fast_mode") else 1024
                 est_tokens = max_tokens
 
                 agent = maybe_create_agent(model_used)
@@ -383,19 +386,10 @@ if generate_now and not st.session_state.get("busy"):
                         system_msg = {"role": "system", "content": prompt_text}
                         user_msg = {"role": "user", "content": "Please summarize the attached video."}
 
-                        response = None
-
                         # normalize response generation across google-generativeai versions
                         response = None
-                        genai.configure(api_key=key_to_use)
-                        fname = file_name_or_id(processed)
-                        if not fname:
-                            raise RuntimeError("Uploaded file missing name/id")
-                        
-                        system_msg = {"role": "system", "content": prompt_text}
-                        user_msg = {"role": "user", "content": "Please summarize the attached video."}
-                        
-                        # 1) responses.generate (modern)
+
+                        # 1) responses.generate
                         try:
                             if hasattr(genai, "responses") and hasattr(genai.responses, "generate"):
                                 response = genai.responses.generate(
@@ -407,8 +401,8 @@ if generate_now and not st.session_state.get("busy"):
                                 )
                         except Exception:
                             response = None
-                        
-                        # 2) GenerativeModel (another modern interface)
+
+                        # 2) GenerativeModel (present in 0.8.5)
                         if response is None:
                             try:
                                 if hasattr(genai, "GenerativeModel"):
@@ -419,8 +413,8 @@ if generate_now and not st.session_state.get("busy"):
                                         response = gm.generate([system_msg, user_msg], files=[{"name": fname}], max_output_tokens=max_tokens)
                             except Exception:
                                 response = None
-                        
-                        # 3) older top-level helpers (legacy)
+
+                        # 3) older top-level helpers
                         if response is None:
                             try:
                                 if hasattr(genai, "generate"):
@@ -429,19 +423,17 @@ if generate_now and not st.session_state.get("busy"):
                                     response = genai.create(model=model_used, input=[{"text": prompt_text, "files": [{"name": fname}]}], max_output_tokens=max_tokens)
                             except Exception:
                                 response = None
-                        
+
                         if response is None:
                             raise RuntimeError("No supported generate method found on google.generativeai; check SDK version or model compatibility.")
-                        
-                        # normalize output extraction robustly
+
                         outputs = []
                         try:
                             if isinstance(response, dict):
-                                # common keys that hold lists
-                                for k in ("output", "candidates", "items", "responses"):
-                                    v = response.get(k)
-                                    if isinstance(v, list) and v:
-                                        outputs = v
+                                for key in ("output", "candidates", "items", "responses"):
+                                    val = response.get(key)
+                                    if isinstance(val, list) and val:
+                                        outputs = val
                                         break
                                 if not outputs:
                                     for v in response.values():
@@ -450,14 +442,13 @@ if generate_now and not st.session_state.get("busy"):
                                             break
                             else:
                                 for attr in ("output", "candidates", "items", "responses"):
-                                    v = getattr(response, attr, None)
-                                    if isinstance(v, (list, tuple)) and v:
-                                        outputs = list(v)
+                                    val = getattr(response, attr, None)
+                                    if isinstance(val, (list, tuple)) and val:
+                                        outputs = list(val)
                                         break
                         except Exception:
                             outputs = []
-                        
-                        # fall back to message/text fields if no list found
+
                         if not outputs:
                             candidate_text = None
                             if isinstance(response, dict):
@@ -466,13 +457,11 @@ if generate_now and not st.session_state.get("busy"):
                                 candidate_text = getattr(response, "text", None) or getattr(response, "message", None)
                             if candidate_text:
                                 outputs = [ {"text": candidate_text} ]
-                        
-                        # extract strings from outputs
+
                         text_pieces = []
                         for item in outputs:
                             if not item:
                                 continue
-                            # dict style
                             if isinstance(item, dict):
                                 for k in ("content", "text", "message", "output_text", "output"):
                                     v = item.get(k)
@@ -489,7 +478,6 @@ if generate_now and not st.session_state.get("busy"):
                                                         text_pieces.append(str(t).strip())
                                         break
                             else:
-                                # object style
                                 for k in ("content", "text", "message", "output", "output_text"):
                                     v = getattr(item, k, None)
                                     if v:
@@ -504,8 +492,7 @@ if generate_now and not st.session_state.get("busy"):
                                                     if t:
                                                         text_pieces.append(str(t).strip())
                                         break
-                        
-                        # dedupe and join
+
                         seen = set()
                         filtered = []
                         for t in text_pieces:
@@ -514,7 +501,6 @@ if generate_now and not st.session_state.get("busy"):
                                 seen.add(t)
                         out = "\n\n".join(filtered)
 
-
                     except Exception as e:
                         tb = traceback.format_exc()
                         st.session_state["last_error"] = f"Responses API error: {e}\n\nDebug: {debug_info}\n\nTraceback:\n{tb}"
@@ -557,8 +543,34 @@ if st.session_state.get("last_error"):
     with st.expander("Last Error", expanded=False):
         st.write(st.session_state.get("last_error"))
 
-import streamlit as st
-from diagnostics import env_check
-
-st.header("Environment")
-st.code(env_check())
+with st.expander("Errors / Environment", expanded=False):
+    try:
+        # diagnostics snippet requested by user, expects diagnostics.env_check() to be present
+        import streamlit as _st  # keep namespace local
+        try:
+            from diagnostics import env_check  # optional helper in user's repo
+            _st.header("Environment")
+            _st.code(env_check())
+        except Exception:
+            # fallback basic env info
+            py = os.popen("python -V").read().strip()
+            try:
+                ver = getattr(genai, "__version__", None)
+                f = getattr(genai, "__file__", None)
+                has_responses = hasattr(genai, "responses")
+                has_GenerativeModel = hasattr(genai, "GenerativeModel")
+                has_generate = hasattr(genai, "generate")
+                info = {
+                    "python_version": py,
+                    "version": ver,
+                    "file": f,
+                    "has_responses": has_responses,
+                    "has_GenerativeModel": has_GenerativeModel,
+                    "has_generate": has_generate,
+                }
+            except Exception:
+                info = {"python_version": py}
+            _st.header("Environment")
+            _st.code(str(info))
+    except Exception:
+        pass