Update app.py

app.py

@@ -40,6 +40,10 @@ query = st.text_input("Search IMDb", "Enter the name of a movie or TV show")
 ds_lang = st.text_input("Default subtitle language (Leave it empty if you do not want to use it)", "")
 sub_url = st.text_input("Subtitle URL (Leave it empty if you do not want to use it)", "")
 
+# Initialize variables
+results = []
+selected_title = None
+
 # Search and display results
 if st.button("Search"):
     if query:
@@ -48,8 +52,10 @@ if st.button("Search"):
         
         # Display search results
         if results:
-            st.write(f"Found {len(results)} results. Displaying up to 20 results:")
-            for idx, result in enumerate(results[:20]):  # Limit to 20 results
+            st.write(f"Found {len(results)} results. Displaying the top 5 results:")
+            titles = []
+            posters = []
+            for idx, result in enumerate(results[:5]):  # Limit to 5 results
                 title = result.get('title', 'Unknown title')
                 year = result.get('year', 'N/A')
                 imdb_id = result.movieID
@@ -58,22 +64,24 @@ if st.button("Search"):
                 ia.update(result, info=['main'])
                 poster_url = result.get('full-size cover url', None)
                 
-                # Display poster, title, and year
-                st.write(f"{idx + 1}. {title} ({year})")
-                if poster_url:
-                    st.image(poster_url, width=150)
-                
-                # Generate embed URL based on search type
-                if search_type == "Movie":
-                    embed_url = get_movie_embed(imdb_id, ds_lang, sub_url)
-                else:
-                    embed_url = get_tv_embed(imdb_id, ds_lang=ds_lang, sub_url=sub_url)
-                
-                # Display the embed URL
-                st.write("URL:")
-                st.write(embed_url)
-                st.write("---")
-        else:
-            st.write("No results found.")
+                # Store title and poster for selection
+                titles.append(f"{title} ({year})")
+                posters.append(poster_url)
+            
+            # Create a selection box for the titles
+            selected_title = st.selectbox("Select a title to play", titles)
+
+# If a title is selected, find the corresponding embed URL and display it
+if selected_title:
+    selected_index = titles.index(selected_title)
+    selected_result = results[selected_index]
+    imdb_id = selected_result.movieID
+    
+    if search_type == "Movie":
+        embed_url = get_movie_embed(imdb_id, ds_lang, sub_url)
     else:
-        st.write("Please enter a search query.")
+        embed_url = get_tv_embed(imdb_id, ds_lang=ds_lang, sub_url=sub_url)
+    
+    # Display the selected movie or TV show using an iframe
+    st.write("Playing:")
+    st.markdown(f'<iframe src="{embed_url}" width="700" height="400" frameborder="0" allowfullscreen></iframe>', unsafe_allow_html=True)