Update streamlit_app.py

streamlit_app.py

@@ -254,14 +254,16 @@ safety_settings = [
 ]
 
 # ---- Upload & processing helpers ----
-def upload_video_sdk(filepath: str):
+def upload_video_sdk(filepath: str, rag_store_name: str = "default_store"):
     key = get_effective_api_key()
     if not key:
         raise RuntimeError("No API key provided")
     if not HAS_GENAI or upload_file is None:
         raise RuntimeError("google.generativeai SDK not available; cannot upload")
     genai.configure(api_key=key)
-    return upload_file(filepath)
+    
+    # Include the ragStoreName parameter
+    return upload_file(filepath, ragStoreName=rag_store_name)
 
 def wait_for_processed(file_obj, timeout: int = None):
     if timeout is None:
@@ -440,6 +442,9 @@ if generate_now and not st.session_state.get("busy"):
                     local_path = current_path
                     upload_path, compressed = compress_video_if_large(local_path, threshold_mb=st.session_state.get("compress_threshold_mb", 200))
 
+                        # Set rag_store_name here
+                    rag_store_name = "your_rag_store_name"  # Adjust this value as needed
+
                     with st.spinner(f"Uploading video{' (compressed)' if compressed else ''}..."):
                         try:
                             uploaded = upload_video_sdk(upload_path)