Update app.py

app.py

@@ -119,7 +119,8 @@ def stream_gemini(
     api_key: str,
 ) -> Generator[str, None, None]:
     """
-    Streams from Google Gemini via google.genai SDK.
+    Streams from Google Gemini via google-genai SDK.
+    Uses the correct streaming interface: client.models.stream_generate_content(...)
     """
     if genai is None:
         yield "❌ Gemini SDK not installed. Add `google-genai` to requirements.txt."
@@ -128,47 +129,34 @@ def stream_gemini(
     client = genai.Client(api_key=api_key)
 
     # Build generation config (Gemini supports these fields; seed is optional)
-    gen_cfg = {
+    cfg_kwargs: Dict[str, Any] = {
         "temperature": float(temperature),
         "top_p": float(top_p),
         "max_output_tokens": int(max_tokens),
     }
     if seed is not None:
-        gen_cfg["seed"] = int(seed)
+        cfg_kwargs["seed"] = int(seed)
 
     response_text = ""
     try:
-        with client.responses.stream(
+        stream = client.models.stream_generate_content(
             model=model,
-            input=prompt,
-            config=genai_types.GenerateContentConfig(
-                temperature=gen_cfg["temperature"],
-                top_p=gen_cfg["top_p"],
-                max_output_tokens=gen_cfg["max_output_tokens"],
-                seed=gen_cfg.get("seed"),
-            ),
-        ) as stream:
-            for event in stream:
-                # Each event may carry incremental text in candidates[0].content.parts[…].text
-                try:
-                    for cand in getattr(event, "candidates", []) or []:
-                        parts = getattr(cand, "content", None)
-                        if parts and getattr(parts, "parts", None):
-                            for p in parts.parts:
-                                txt = getattr(p, "text", None)
-                                if txt:
-                                    response_text += txt
-                                    yield response_text
-                except Exception:
-                    # Best-effort incremental parse
-                    pass
-
-            # Ensure final text is yielded (for some drivers, the last event is summary)
-            final = getattr(stream, "text", None)
-            if final and final not in response_text:
-                response_text += final
+            contents=prompt,
+            config=genai_types.GenerateContentConfig(**cfg_kwargs),
+        )
+        for chunk in stream:
+            # Each chunk typically exposes incremental text as .text
+            txt = getattr(chunk, "text", None)
+            if txt:
+                response_text += txt
                 yield response_text
 
+        # Some drivers may finish with a final aggregate text
+        final = getattr(stream, "text", None)
+        if final and final not in response_text:
+            response_text += final
+            yield response_text
+
     except Exception as e:
         yield f"❌ Gemini error: {type(e).__name__}: {e}"