Upload app.py

app.py

@@ -0,0 +1,142 @@
+import streamlit as st
+from imdb import IMDb
+import requests
+from PIL import Image
+from io import BytesIO
+
+# Function to search for movies/series
+def search_movies(movie_name):
+    ia = IMDb()
+    search_results = ia.search_movie(movie_name)
+    return search_results
+
+# Function to fetch image from URL
+def fetch_image(url):
+    response = requests.get(url)
+    img = Image.open(BytesIO(response.content))
+    return img
+
+# Function to get movie details
+def get_movie_details(imdb_id):
+    ia = IMDb()
+    movie = ia.get_movie(imdb_id)
+    details = {
+        'title': movie['title'],
+        'year': movie.get('year'),
+        'rating': movie.get('rating'),
+        'genres': movie.get('genres'),
+        'plot': movie.get('plot'),
+        'poster_url': movie.get('full-size cover url'),
+        'kind': movie.get('kind'),
+        'episodes': movie.get('episodes') if movie.get('kind') == 'tv series' else None
+    }
+    return details
+
+def main():
+    st.set_page_config(page_title="Yuyutsu MovieDrive", page_icon="🦇", layout="wide")
+    st.title("Yuyutsu MovieDrive")
+
+    # Initialize session state
+    if 'search_results' not in st.session_state:
+        st.session_state.search_results = None
+    if 'selected_movie' not in st.session_state:
+        st.session_state.selected_movie = None
+    if 'details' not in st.session_state:
+        st.session_state.details = None
+    if 'imdb_id' not in st.session_state:
+        st.session_state.imdb_id = None
+
+    movie_name = st.text_input("Enter the movie or series name:")
+    is_webseries = st.checkbox("This is a web series")
+
+    if movie_name and st.button("Search"):
+        st.session_state.search_results = search_movies(movie_name)
+        st.session_state.selected_movie = None
+        st.session_state.details = None
+        st.session_state.imdb_id = None
+
+    if st.session_state.search_results:
+        st.write(f"Found {len(st.session_state.search_results)} results. Select one:")
+        selected_movie = st.selectbox(
+            "Select a movie or series:",
+            options=[movie["title"] for movie in st.session_state.search_results],
+            key="movie_select"
+        )
+
+        if selected_movie != st.session_state.selected_movie:
+            st.session_state.selected_movie = selected_movie
+            movie_data = next(m for m in st.session_state.search_results if m['title'] == selected_movie)
+            st.session_state.imdb_id = movie_data.movieID
+            st.session_state.details = get_movie_details(st.session_state.imdb_id)
+
+    if st.session_state.details and st.session_state.imdb_id:
+        details = st.session_state.details
+        imdb_id = st.session_state.imdb_id
+
+        col1, col2 = st.columns([1, 2])
+        with col1:
+            if details.get('poster_url'):
+                poster = fetch_image(details['poster_url'])
+                st.image(poster, caption=details['title'], use_column_width=True)
+
+        with col2:
+            st.subheader(f"{details['title']} ({details.get('year', 'N/A')})")
+            st.write(f"**IMDb Rating:** {details.get('rating', 'Not rated')}")
+            st.write(f"**Genres:** {', '.join(details.get('genres', ['Not specified']))}")
+            st.write(f"**Plot:** {details.get('plot', 'No plot available')}")
+
+        st.subheader("Watch Options")
+
+        # VidSrc link
+        if is_webseries or details['kind'] == 'tv series':
+            vidsrc_url = f"https://vidsrc.xyz/embed/tv?imdb=tt{imdb_id}"
+            st.write(f"Click the link below to watch the web series on vidsrc:")
+            st.markdown(f"[Watch {details['title']} on vidsrc]({vidsrc_url})")
+        else:
+            vidsrc_url = f"https://vidsrc.xyz/embed/movie?imdb=tt{imdb_id}"
+            st.write(f"Click the link below to watch on vidsrc:")
+            st.markdown(f"[Watch {details['title']} on vidsrc]({vidsrc_url})")
+
+        # Autoembed link
+        if is_webseries or details['kind'] == 'tv series':
+            st.write("For TV Series, select season and episode:")
+            
+            if details['episodes']:
+                seasons = sorted(details['episodes'].keys())
+                season = st.selectbox("Season", options=seasons)
+                
+                if season:
+                    episodes = sorted(details['episodes'][season].keys())
+                    episode = st.selectbox("Episode", options=episodes)
+                    
+                    if episode:
+                        autoembed_url = f"https://player.autoembed.cc/embed/tv/tt{imdb_id}/{season}/{episode}"
+                        st.markdown(f"[Watch {details['title']} S{season:02d}E{episode:02d} on autoembed]({autoembed_url})")
+            else:
+                st.write("Episode information is not available for this series.")
+                season = st.number_input("Enter season number:", min_value=1, value=1)
+                episode = st.number_input("Enter episode number:", min_value=1, value=1)
+                autoembed_url = f"https://player.autoembed.cc/embed/tv/tt{imdb_id}/{season}/{episode}"
+                st.markdown(f"[Watch {details['title']} S{season:02d}E{episode:02d} on autoembed]({autoembed_url})")
+
+        elif details['kind'] == 'movie':
+            autoembed_url = f"https://player.autoembed.cc/embed/movie/tt{imdb_id}"
+            st.write(f"Alternative link for movies:")
+            st.markdown(f"[Watch {details['title']} on autoembed]({autoembed_url})")
+        else:
+            st.write("Note: Alternative link is not available for this type of content.")
+
+        st.write("Note: These links will open in a new tab. Please ensure you have appropriate permissions to view the content.")
+
+    # Add a footer
+    st.markdown("---")
+    st.markdown(
+        "<h3 style='text-align: center;'>"
+        "Created with ❤️ by Yuyutsu<br>"
+        "<span style='color: #FF9933;'>जय श्री राम</span>"
+        "</h3>", 
+        unsafe_allow_html=True
+    )
+
+if __name__ == "__main__":
+    main()