Update app.py

app.py

@@ -26,10 +26,13 @@ class VideoProcessor:
         except Exception as e:
             raise RuntimeError(f"Failed to upload video: {str(e)}")
 
-    def wait_for_processing(self, video_file):
+    def wait_for_processing(self, video_file, status_placeholder):
         max_attempts = 60  # Maximum wait time: 2 minutes
         attempts = 0
         while video_file.state.name == "PROCESSING" and attempts < max_attempts:
+            # Update status text with dots animation
+            dots = "." * ((attempts % 3) + 1)
+            status_placeholder.markdown(f"**Processing video{dots}**")
             time.sleep(2)
             video_file = genai.get_file(video_file.name)
             attempts += 1
@@ -56,6 +59,41 @@ if "video_file" not in st.session_state:
 if "video_name" not in st.session_state:
     st.session_state.video_name = None
 
+# Buffering animation CSS
+def show_buffering_animation():
+    st.markdown("""
+        <style>
+        .buffering-container {
+            display: flex;
+            flex-direction: column;
+            align-items: center;
+            justify-content: center;
+            padding: 20px;
+        }
+        .buffering-spinner {
+            width: 50px;
+            height: 50px;
+            border: 5px solid #f3f3f3;
+            border-top: 5px solid #3498db;
+            border-radius: 50%;
+            animation: spin 1s linear infinite;
+        }
+        @keyframes spin {
+            0% { transform: rotate(0deg); }
+            100% { transform: rotate(360deg); }
+        }
+        .buffering-text {
+            margin-top: 10px;
+            font-size: 16px;
+            color: #666;
+        }
+        </style>
+        <div class="buffering-container">
+            <div class="buffering-spinner"></div>
+            <div class="buffering-text">Processing video...</div>
+        </div>
+    """, unsafe_allow_html=True)
+
 # Main app function
 def main():
     st.set_page_config(page_title="Video Retrieval-Augmented Generation", layout="wide")
@@ -104,25 +142,32 @@ def main():
                         tmp.write(uploaded_file.getvalue())
                         tmp_path = tmp.name
 
-                    # Upload and process with progress indication
-                    with st.spinner("Uploading and processing video..."):
-                        progress_bar = st.progress(0)
-                        progress_bar.progress(25, text="Uploading video...")
-                        
-                        video_file = st.session_state.video_processor.upload_video(tmp_path, uploaded_file.name)
-                        progress_bar.progress(50, text="Processing video...")
-                        
-                        processed_file = st.session_state.video_processor.wait_for_processing(video_file)
-                        progress_bar.progress(100, text="Complete!")
-                        
-                        # Update session state
-                        st.session_state.video_file = processed_file
-                        st.session_state.video_name = uploaded_file.name
-                        st.session_state.messages = []  # Clear previous conversation
-                        
-                        st.success("Video processed successfully!")
-                        time.sleep(1)  # Show success message briefly
-                        progress_bar.empty()  # Clear progress bar
+                    # Show buffering animation container
+                    buffering_container = st.empty()
+                    status_text = st.empty()
+                    
+                    with buffering_container.container():
+                        show_buffering_animation()
+                    
+                    # Upload video
+                    status_text.markdown("**Uploading video...**")
+                    video_file = st.session_state.video_processor.upload_video(tmp_path, uploaded_file.name)
+                    
+                    # Process video with status updates
+                    processed_file = st.session_state.video_processor.wait_for_processing(video_file, status_text)
+                    
+                    # Clear buffering animation
+                    buffering_container.empty()
+                    status_text.empty()
+                    
+                    # Update session state
+                    st.session_state.video_file = processed_file
+                    st.session_state.video_name = uploaded_file.name
+                    st.session_state.messages = []  # Clear previous conversation
+                    
+                    st.success("Video processed successfully!")
+                    time.sleep(1)  # Show success message briefly
+                    st.rerun()
 
                 except Exception as e:
                     st.error(f"Error processing video: {str(e)}")