Update app.py

app.py

@@ -40,10 +40,6 @@ 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:
@@ -52,10 +48,8 @@ if st.button("Search"):
         
         # Display search results
         if 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
+            st.write(f"Found {len(results)} results. Displaying up to 20 results:")
+            for idx, result in enumerate(results[:5]):  # Limit to 20 results
                 title = result.get('title', 'Unknown title')
                 year = result.get('year', 'N/A')
                 imdb_id = result.movieID
@@ -64,24 +58,22 @@ if st.button("Search"):
                 ia.update(result, info=['main'])
                 poster_url = result.get('full-size cover url', None)
                 
-                # 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)
+                # 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.")
     else:
-        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)
+        st.write("Please enter a search query.")