Update app.py

app.py

@@ -220,10 +220,37 @@ def page_1():
             progress_bar.progress(100)
             status_text.text("Processing completed successfully!")
             st.success("PDF processed successfully! Collections saved to session state.")
+
         except Exception as e:
             progress_bar.progress(0)
             status_text.text("")
             st.error(f"Error processing PDF: {e}")
+        
+        # Simple Video Query Section
+        st.subheader("Search for Videos")
+
+        # Query input for videos
+        video_query = st.text_input("Enter a query to search videos", value="")
+        
+        if video_query:
+            video_results = video_collection.query(
+                query_texts=[video_query],
+                n_results=5,
+                include=['uris', 'distances', 'metadatas']
+            )
+
+            uris = video_results['uris'][0]
+            distances = video_results['distances'][0]
+            metadatas = video_results['metadatas'][0]
+
+            if uris:
+                st.write(f"Found {len(uris)} video(s) for your query.")
+                for uri, distance, metadata in zip(uris, distances, metadatas):
+                    video_uri = metadata['video_uri']
+                    st.write(f"Video URI: {video_uri} (Distance: {distance})")
+                    st.video(video_uri)  # Display video
+            else:
+                st.write("No videos found for the query.")
 
 
 def page_2():