Update streamlit_app.py

streamlit_app.py

@@ -135,20 +135,28 @@ def strip_prompt_echo(prompt: str, text: str) -> str:
 
 
 def generate_inline(video_path: str, prompt: str, model_id: str, timeout: int) -> str:
+    # 1️⃣ read video and encode as base64
     with open(video_path, "rb") as f:
         b64 = base64.b64encode(f.read()).decode()
+
+    # 2️⃣ build the “inline” part that the Gemini API expects
     video_part = {
         "inline_data": {"mime_type": "video/mp4", "data": b64}
     }
     contents = [prompt, video_part]
 
-    client = genai.Client()
-    resp = client.models.generate_content(
-        model=model_id,
-        contents=contents,
+    # 3️⃣ create a GenerativeModel object (no Client class any more)
+    model = genai.GenerativeModel(model_name=model_id)
+
+    # 4️⃣ call generate_content – the timeout is passed via the request options
+    resp = model.generate_content(
+        contents,
         generation_config={"max_output_tokens": 1024},
-        timeout=timeout,
+        # `timeout` is a kw‑arg of the underlying HTTP request; the SDK forwards it
+        request_options={"timeout": timeout},
     )
+
+    # 5️⃣ the response object has a `.text` attribute (or `.parts` for multi‑part)
     return getattr(resp, "text", str(resp))