Create app.py

app.py

@@ -0,0 +1,75 @@
+import streamlit as st
+from imdb import IMDb
+
+# Initialize IMDb object
+ia = IMDb()
+
+# Function to get movie embed URL
+def get_movie_embed(imdb_id, ds_lang=None, sub_url=None):
+    base_url = "https://vidsrc.xyz/embed/movie"
+    params = f"?imdb={imdb_id}"
+    if ds_lang:
+        params += f"&ds_lang={ds_lang}"
+    if sub_url:
+        params += f"&sub_url={sub_url}"
+    return f"{base_url}{params}"
+
+# Function to get TV show embed URL
+def get_tv_embed(imdb_id, season=None, episode=None, ds_lang=None, sub_url=None):
+    base_url = "https://vidsrc.xyz/embed/tv"
+    if season and episode:
+        return f"{base_url}/{imdb_id}/{season}-{episode}"
+    params = f"?imdb={imdb_id}"
+    if ds_lang:
+        params += f"&ds_lang={ds_lang}"
+    if sub_url:
+        params += f"&sub_url={sub_url}"
+    return f"{base_url}{params}"
+
+# Streamlit app layout
+st.title("Vidsrc Embed URL Generator")
+st.write("Search for a movie or TV show using IMDbPY and get the Vidsrc embed URL.")
+
+# Input for search type
+search_type = st.radio("Select type", ("Movie", "TV Show"))
+
+# Input for search query
+query = st.text_input("Search IMDb", "Enter the name of a movie or TV show")
+
+# Optional inputs for additional parameters
+ds_lang = st.text_input("Default subtitle language (ISO639 code)", "")
+sub_url = st.text_input("Subtitle URL (with CORS enabled)", "")
+
+# Search and display results
+if st.button("Search"):
+    if query:
+        # Perform search
+        results = ia.search_movie(query)
+        
+        # Display search results
+        if results:
+            st.write(f"Found {len(results)} results.")
+            for idx, result in enumerate(results):
+                st.write(f"{idx + 1}. {result['title']} ({result.get('year', 'N/A')})")
+
+            # Input for selection
+            selection = st.number_input("Select the number of the result", min_value=1, max_value=len(results))
+
+            selected_result = results[selection - 1]
+            imdb_id = selected_result.movieID
+
+            # Generate embed URL based on search type
+            if search_type == "Movie":
+                embed_url = get_movie_embed(imdb_id, ds_lang, sub_url)
+            else:
+                season = st.number_input("Season number", min_value=1)
+                episode = st.number_input("Episode number", min_value=1)
+                embed_url = get_tv_embed(imdb_id, season, episode, ds_lang, sub_url)
+
+            # Display the embed URL
+            st.write("Embed URL:")
+            st.write(embed_url)
+        else:
+            st.write("No results found.")
+    else:
+        st.write("Please enter a search query.")