Update app.py

app.py

@@ -6,102 +6,91 @@ import time
 import mimetypes
 from pathlib import Path
 
-# Hard-coded Gemini API key
+# Gemini API key (hardcoded for demo purposes)
 GEMINI_API_KEY = "AIzaSyDCMPwXHagWqYTQB3HL7FceHEmKUv3v4wc"
 genai.configure(api_key=GEMINI_API_KEY)
 
-def main():
-    # Set Streamlit page configuration
-    st.set_page_config(
-        page_title="Video RAG with Gemini",
-        page_icon="🎬",
-        layout="wide"
-    )
-
-    class VideoProcessor:
-        def __init__(self):
-            self.model = genai.GenerativeModel("gemini-2.0-flash")
-
-        def upload_video(self, video_path, display_name="uploaded_video"):
-            return genai.upload_file(path=video_path, display_name=display_name)
-
-        def wait_for_processing(self, video_file):
-            while video_file.state.name == "PROCESSING":
-                time.sleep(2)
-                video_file = genai.get_file(video_file.name)
-            if video_file.state.name == "FAILED":
-                raise RuntimeError("Video processing failed")
-            return video_file
-
-        def chat_with_video(self, video_file, prompt):
-            response = self.model.generate_content([video_file, prompt])
-            return response.text
-
-    # Initialize session state
-    for key in ["video_processor", "video_file", "video_name", "messages"]:
-        if key not in st.session_state:
-            st.session_state[key] = None if key != "messages" else []
-
-    # Sidebar UI
-    with st.sidebar:
-        st.header("📹 Upload Video")
-        uploaded_file = st.file_uploader("Choose video", type=['mp4', 'mov', 'avi', 'mkv', 'webm'])
-
-        if uploaded_file:
-            if mimetypes.guess_type(uploaded_file.name)[0].startswith("video"):
-                file_size = len(uploaded_file.getvalue()) / (1024**2)
-                st.info(f"Size: {file_size:.2f} MB")
-
-                if st.session_state.video_name != uploaded_file.name:
-                    st.session_state.video_processor = VideoProcessor()
-                    with tempfile.NamedTemporaryFile(delete=False, suffix=Path(uploaded_file.name).suffix) as tmp:
-                        tmp.write(uploaded_file.getvalue())
-                        tmp_path = tmp.name
-
-                    with st.spinner("Uploading and processing..."):
-                        video_file = st.session_state.video_processor.upload_video(tmp_path, uploaded_file.name)
-                        processed_file = st.session_state.video_processor.wait_for_processing(video_file)
-                        st.session_state.video_file = processed_file
-                        st.session_state.video_name = uploaded_file.name
-                        st.session_state.messages.clear()
-                        st.success("✅ Video processed")
-
-                    os.unlink(tmp_path)
-
-                st.video(uploaded_file.getvalue())
-            else:
-                st.error("Not a valid video file")
-
-        if st.button("Reset Chat"):
-            st.session_state.messages.clear()
-
-        if st.button("Reset All"):
-            st.session_state.clear()
-
-    # Main Chat UI
-    st.title("🎬 Video RAG with Gemini")
-
-    if not st.session_state.video_file:
-        st.info("👈 Upload a video to start chatting")
-    else:
-        for msg in st.session_state.messages:
-            with st.chat_message(msg["role"]):
-                st.markdown(msg["content"])
-
-        prompt = st.chat_input("Ask about the video...")
-
-        if prompt:
-            st.session_state.messages.append({"role": "user", "content": prompt})
-            with st.chat_message("user"):
-                st.markdown(prompt)
-
-            with st.chat_message("assistant"):
-                placeholder = st.empty()
-                with st.spinner("Thinking..."):
-                    response = st.session_state.video_processor.chat_with_video(st.session_state.video_file, prompt)
-
-                placeholder.markdown(response)
-                st.session_state.messages.append({"role": "assistant", "content": response})
-
-if __name__ == "__main__":
-    main()
+st.set_page_config(page_title="Video RAG with Gemini", page_icon="🎬", layout="wide")
+
+class VideoProcessor:
+    def __init__(self):
+        self.model = genai.GenerativeModel("gemini-2.0-flash")
+
+    def upload_video(self, video_path, display_name="uploaded_video"):
+        return genai.upload_file(path=video_path, display_name=display_name)
+
+    def wait_for_processing(self, video_file):
+        while video_file.state.name == "PROCESSING":
+            time.sleep(2)
+            video_file = genai.get_file(video_file.name)
+        if video_file.state.name == "FAILED":
+            raise RuntimeError("Video processing failed")
+        return video_file
+
+    def chat_with_video(self, video_file, prompt):
+        response = self.model.generate_content([video_file, prompt])
+        return response.text
+
+# Initialize session state
+for key in ["video_processor", "video_file", "video_name", "messages"]:
+    if key not in st.session_state:
+        st.session_state[key] = None if key != "messages" else []
+
+with st.sidebar:
+    st.header("📹 Upload Video")
+    uploaded_file = st.file_uploader("Choose video", type=['mp4', 'mov', 'avi', 'mkv', 'webm'])
+
+    if uploaded_file:
+        if mimetypes.guess_type(uploaded_file.name)[0].startswith("video"):
+            file_size = len(uploaded_file.getvalue()) / (1024**2)
+            st.info(f"Size: {file_size:.2f} MB")
+
+            if st.session_state.video_name != uploaded_file.name:
+                st.session_state.video_processor = VideoProcessor()
+                with tempfile.NamedTemporaryFile(delete=False, suffix=Path(uploaded_file.name).suffix) as tmp:
+                    tmp.write(uploaded_file.getvalue())
+                    tmp_path = tmp.name
+
+                with st.spinner("Uploading and processing..."):
+                    video_file = st.session_state.video_processor.upload_video(tmp_path, uploaded_file.name)
+                    processed_file = st.session_state.video_processor.wait_for_processing(video_file)
+                    st.session_state.video_file = processed_file
+                    st.session_state.video_name = uploaded_file.name
+                    st.session_state.messages.clear()
+                    st.success("✅ Video processed")
+
+                os.unlink(tmp_path)
+
+            st.video(uploaded_file.getvalue())
+        else:
+            st.error("Not a valid video file")
+
+    if st.button("Reset Chat"):
+        st.session_state.messages.clear()
+
+    if st.button("Reset All"):
+        st.session_state.clear()
+
+st.title("🎬 Video RAG with Gemini")
+
+if not st.session_state.video_file:
+    st.info("👈 Upload a video to start chatting")
+else:
+    for msg in st.session_state.messages:
+        with st.chat_message(msg["role"]):
+            st.markdown(msg["content"])
+
+    prompt = st.chat_input("Ask about the video...")
+
+    if prompt:
+        st.session_state.messages.append({"role": "user", "content": prompt})
+        with st.chat_message("user"):
+            st.markdown(prompt)
+
+        with st.chat_message("assistant"):
+            placeholder = st.empty()
+            with st.spinner("Thinking..."):
+                response = st.session_state.video_processor.chat_with_video(st.session_state.video_file, prompt)
+
+            placeholder.markdown(response)
+            st.session_state.messages.append({"role": "assistant", "content": response})