Update app.py

app.py

@@ -1,51 +1,129 @@
 import gradio as gr
 import pandas as pd
+import requests
+from typing import List, Dict, Any
+import os
 
-# Sample data with one actual movie
-movie_data = {
-    "title": ["Big Buck Bunny"],
-    "poster_url": [
-        "https://upload.wikimedia.org/wikipedia/commons/thumb/c/c5/Big_buck_bunny_poster_big.jpg/800px-Big_buck_bunny_poster_big.jpg"
-    ],
-    "movie_url": [
-        "https://storage.googleapis.com/gtv-videos-bucket/sample/BigBuckBunny.mp4"  # Open source test video
-    ],
-    "genre": ["Animation"],
-    "rating": [4.5]
-}
+# TMDB API Configuration (Environment Variables)
+TMDB_API_KEY = os.environ.get("TMDB_API_KEY", "YOUR_API_KEY")
+TMDB_API_TOKEN = os.environ.get("TMDB_API_TOKEN", "YOUR_API_TOKEN")
+TMDB_BASE_URL = "https://api.themoviedb.org/3"
+TMDB_IMAGE_BASE_URL = "https://image.tmdb.org/t/p/w500"
 
-content_df = pd.DataFrame(movie_data)
+def fetch_genres() -> List[Dict[str, Any]]:
+    """Fetch movie genres from TMDB"""
+    headers = {
+        "Authorization": f"Bearer {TMDB_API_TOKEN}",
+        "accept": "application/json"
+    }
+    response = requests.get(f"{TMDB_BASE_URL}/genre/movie/list", headers=headers)
+    return response.json()["genres"]
 
-def list_movies(selected_genre):
-    if selected_genre != "All":
-        filtered_movies = content_df[content_df["genre"] == selected_genre]
-    else:
-        filtered_movies = content_df
+def fetch_movies_by_genre(genre_id: str = None) -> pd.DataFrame:
+    """Fetch movies from TMDB, optionally filtered by genre"""
+    headers = {
+        "Authorization": f"Bearer {TMDB_API_TOKEN}",
+        "accept": "application/json"
+    }
     
-    return filtered_movies["poster_url"].tolist()
+    params = {
+        "include_adult": False,
+        "include_video": True,
+        "language": "en-US",
+        "page": 1,
+        "sort_by": "popularity.desc"
+    }
+    
+    if genre_id and genre_id != "All":
+        params["with_genres"] = genre_id
+    
+    response = requests.get(f"{TMDB_BASE_URL}/discover/movie", headers=headers, params=params)
+    movies = response.json()["results"]
+    
+    # Convert to DataFrame
+    movie_data = {
+        "title": [],
+        "poster_url": [],
+        "overview": [],
+        "rating": [],
+        "id": []
+    }
+    
+    for movie in movies:
+        movie_data["title"].append(movie["title"])
+        movie_data["poster_url"].append(
+            f"{TMDB_IMAGE_BASE_URL}{movie['poster_path']}" if movie['poster_path'] else None
+        )
+        movie_data["overview"].append(movie["overview"])
+        movie_data["rating"].append(movie["vote_average"])
+        movie_data["id"].append(movie["id"])
+    
+    return pd.DataFrame(movie_data)
+
+def get_movie_trailer(movie_id: int) -> str:
+    """Fetch movie trailer URL from TMDB"""
+    headers = {
+        "Authorization": f"Bearer {TMDB_API_TOKEN}",
+        "accept": "application/json"
+    }
+    
+    response = requests.get(
+        f"{TMDB_BASE_URL}/movie/{movie_id}/videos",
+        headers=headers
+    )
+    
+    videos = response.json().get("results", [])
+    trailer = next(
+        (v for v in videos if v["type"] == "Trailer" and v["site"] == "YouTube"),
+        None
+    )
+    
+    if trailer:
+        return f"https://www.youtube.com/watch?v={trailer['key']}"
+    return None
 
-def get_movie_info(evt: gr.SelectData):
+# Initialize global variables
+genres = fetch_genres()
+genre_map = {str(g["id"]): g["name"] for g in genres}
+genre_map["All"] = "All"
+content_df = fetch_movies_by_genre()
+
+def list_movies(selected_genre: str) -> List[str]:
+    """Update movie list based on selected genre"""
+    global content_df
+    content_df = fetch_movies_by_genre(selected_genre if selected_genre != "All" else None)
+    return content_df["poster_url"].tolist()
+
+def get_movie_info(evt: gr.SelectData) -> tuple:
+    """Get selected movie information"""
     selected_idx = evt.index
     if selected_idx >= len(content_df):
-        return None, None
+        return None, None, None
     
     selected_movie = content_df.iloc[selected_idx]
-    return selected_movie["movie_url"], selected_movie["title"]
+    trailer_url = get_movie_trailer(selected_movie["id"])
+    
+    return (
+        trailer_url,
+        selected_movie["title"],
+        f"{selected_movie['title']}\n\nRating: {selected_movie['rating']}/10\n\n{selected_movie['overview']}"
+    )
 
-def rate_movie(movie_title, rating):
+def rate_movie(movie_title: str, rating: float) -> str:
+    """Handle movie rating submission"""
     if not movie_title:
         return "Please select a movie first!"
     
     content_df.loc[content_df["title"] == movie_title, "rating"] = rating
-    return f"Thank you for rating {movie_title}! New rating: {rating}"
+    return f"Thank you for rating {movie_title}! New rating: {rating}/5"
 
 # Gradio UI
-with gr.Blocks() as movie_app:
+with gr.Blocks(theme=gr.themes.Soft()) as movie_app:
     gr.Markdown("# 🎬 Movie Streaming App")
     
     with gr.Row():
         genre_dropdown = gr.Dropdown(
-            choices=["All"] + content_df["genre"].unique().tolist(),
+            choices=list(genre_map.items()),
             value="All",
             label="Filter by Genre"
         )
@@ -55,18 +133,23 @@ with gr.Blocks() as movie_app:
             value=content_df["poster_url"].tolist(),
             label="Movie Gallery",
             show_label=True,
-            columns=[1],
-            height="auto"
+            columns=[4],
+            height="auto",
+            allow_preview=False
         )
     
     with gr.Row():
-        video_player = gr.Video(label="Now Playing")
+        video_player = gr.Video(label="Trailer")
     
     with gr.Row():
-        selected_movie_title = gr.Textbox(label="Selected Movie", interactive=False)
-        rating_input = gr.Slider(minimum=1, maximum=5, step=0.1, label="Rate This Movie")
-        rate_button = gr.Button("Submit Rating")
-        rating_output = gr.Textbox(label="Rating Status")
+        with gr.Column():
+            selected_movie_title = gr.Textbox(label="Selected Movie", interactive=False)
+            movie_info = gr.TextArea(label="Movie Information", interactive=False)
+            
+        with gr.Column():
+            rating_input = gr.Slider(minimum=1, maximum=5, step=0.5, label="Rate This Movie")
+            rate_button = gr.Button("Submit Rating")
+            rating_output = gr.Textbox(label="Rating Status")
 
     # Event handlers
     genre_dropdown.change(
@@ -77,7 +160,7 @@ with gr.Blocks() as movie_app:
     
     gallery.select(
         fn=get_movie_info,
-        outputs=[video_player, selected_movie_title]
+        outputs=[video_player, selected_movie_title, movie_info]
     )
     
     rate_button.click(
@@ -87,4 +170,5 @@ with gr.Blocks() as movie_app:
     )
 
 # Launch app
-movie_app.launch()
+if __name__ == "__main__":
+    movie_app.launch(share=True)