sps / app.py
slimshadow's picture
Create app.py
5a55e71 verified
raw
history blame
1.93 kB
import streamlit as st
import requests
# Function to search for movies or TV shows using the TMDb API
def search_tmdb(query, media_type, api_key):
url = f"https://api.themoviedb.org/3/search/{media_type}"
params = {
"api_key": api_key,
"query": query,
"language": "en-US",
"page": 1,
}
response = requests.get(url, params=params)
results = response.json().get("results", [])
return results[:5] # Return only the first 5 results
# Function to generate the embed link
def generate_embed_link(tmdb_id, media_type):
base_url = "https://v1.sdsp.xyz/embed"
if media_type == "movie":
return f"{base_url}/movies/{tmdb_id}"
elif media_type == "tv":
return f"{base_url}/tv/{tmdb_id}"
# Streamlit app layout
st.title("TMDb Movie & TV Show Search")
st.write("Search for a movie or TV show and get 5 results with TMDb ID and embed link.")
# Input fields for search query and media type
query = st.text_input("Enter the movie or TV show name:")
media_type = st.selectbox("Select media type:", ["movie", "tv"])
# API key (replace 'your_api_key' with your actual TMDb API key)
api_key = "c8402e1c7b8bb9bcbf95b3fa3bad4d84"
if st.button("Search"):
if query and api_key:
results = search_tmdb(query, media_type, api_key)
if results:
for result in results:
title = result["title"] if media_type == "movie" else result["name"]
tmdb_id = result["id"]
embed_link = generate_embed_link(tmdb_id, media_type)
st.write(f"**Title:** {title}")
st.write(f"**TMDb ID:** {tmdb_id}")
st.write(f"**Embed Link:** [Watch here]({embed_link})")
st.write("---")
else:
st.write("No results found. Please try a different query.")
else:
st.write("Please enter a search query and ensure the API key is set.")