Update app.py

app.py

@@ -1,46 +1,28 @@
+import os
 import gradio as gr
 import pandas as pd
 import requests
 from typing import List, Dict, Any
-import os
-import time  # For rate limiting
 
-# 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 API Configuration - Using environment variables for security
+TMDB_API_KEY = os.getenv("TMDB_API_KEY", "2818694258d166d0644b0f75a1b93e59")
+TMDB_API_TOKEN = os.getenv("TMDB_API_TOKEN", "eyJhbGciOiJIUzI1NiJ9.eyJhdWQiOiIyODE4Njk0MjU4ZDE2NmQwNjQ0YjBmNzVhMWI5M2U1OSIsIm5iZiI6MTczMDYxNjE4Mi4xODY5MDI4LCJzdWIiOiI2NzI3MWExYjM5OGM5MDQzZjgwZDEwOGEiLCJzY29wZXMiOlsiYXBpX3JlYWQiXSwidmVyc2lvbiI6MX0.G-pLEQKK3o5ISpb0CEztamPFGlsmpnPqPagawMve9dY")
 TMDB_BASE_URL = "https://api.themoviedb.org/3"
 TMDB_IMAGE_BASE_URL = "https://image.tmdb.org/t/p/w500"
 
-def make_api_request(url, headers, params=None, max_retries=3, delay=2):
-    """Makes a request to the TMDb API with retries and rate limiting."""
-    retries = 0
-    while retries < max_retries:
-        try:
-            response = requests.get(url, headers=headers, params=params)
-            response.raise_for_status()  # Raise an exception for HTTP errors
-            return response
-        except requests.exceptions.RequestException as e:
-            print(f"API request failed: {e}")
-            retries += 1
-            if retries < max_retries:
-                print(f"Retrying in {delay} seconds...")
-                time.sleep(delay)
-    return None
-
 def fetch_genres() -> List[Dict[str, Any]]:
     """Fetch movie genres from TMDB"""
     headers = {
         "Authorization": f"Bearer {TMDB_API_TOKEN}",
         "accept": "application/json"
     }
-    response = make_api_request(f"{TMDB_BASE_URL}/genre/movie/list", headers=headers)
-    if response:
-        try:
-            return response.json()["genres"]  # Handle potential errors
-        except KeyError:
-            print("Error: 'genres' key not found in the TMDb response.")
-            return []
-    return []
+    try:
+        response = requests.get(f"{TMDB_BASE_URL}/genre/movie/list", headers=headers)
+        response.raise_for_status()
+        return response.json()["genres"]
+    except Exception as e:
+        print(f"Error fetching genres: {e}")
+        return []
 
 def fetch_movies_by_genre(genre_id: str = None) -> pd.DataFrame:
     """Fetch movies from TMDB, optionally filtered by genre"""
@@ -60,31 +42,32 @@ def fetch_movies_by_genre(genre_id: str = None) -> pd.DataFrame:
     if genre_id and genre_id != "All":
         params["with_genres"] = genre_id
     
-    response = make_api_request(f"{TMDB_BASE_URL}/discover/movie", headers=headers, params=params)
-    if response:
+    try:
+        response = requests.get(f"{TMDB_BASE_URL}/discover/movie", headers=headers, params=params)
+        response.raise_for_status()
         movies = response.json()["results"]
-    else:
-        movies = []
-
-    # 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)
+        
+        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.get('poster_path') else None
+            )
+            movie_data["overview"].append(movie.get("overview", "No overview available"))
+            movie_data["rating"].append(movie.get("vote_average", 0))
+            movie_data["id"].append(movie["id"])
+        
+        return pd.DataFrame(movie_data)
+    except Exception as e:
+        print(f"Error fetching movies: {e}")
+        return pd.DataFrame(columns=["title", "poster_url", "overview", "rating", "id"])
 
 def get_movie_trailer(movie_id: int) -> str:
     """Fetch movie trailer URL from TMDB"""
@@ -93,24 +76,24 @@ def get_movie_trailer(movie_id: int) -> str:
         "accept": "application/json"
     }
     
-    response = make_api_request(
-        f"{TMDB_BASE_URL}/movie/{movie_id}/videos",
-        headers=headers
-    )
-    
-    if response:
+    try:
+        response = requests.get(
+            f"{TMDB_BASE_URL}/movie/{movie_id}/videos",
+            headers=headers
+        )
+        response.raise_for_status()
         videos = response.json().get("results", [])
-    else:
-        videos = []
-
-    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
+        trailer = next(
+            (v for v in videos if v["type"] == "Trailer" and v["site"] == "YouTube"),
+            None
+        )
+        
+        if trailer:
+            return f'<iframe width="100%" height="400" src="https://www.youtube.com/embed/{trailer["key"]}" frameborder="0" allowfullscreen></iframe>'
+        return '<p>No trailer available</p>'
+    except Exception as e:
+        print(f"Error fetching trailer: {e}")
+        return '<p>Error loading trailer</p>'
 
 # Initialize global variables
 genres = fetch_genres()
@@ -131,10 +114,10 @@ def get_movie_info(evt: gr.SelectData) -> tuple:
         return None, None, None
     
     selected_movie = content_df.iloc[selected_idx]
-    trailer_url = get_movie_trailer(selected_movie["id"])
+    trailer_html = get_movie_trailer(selected_movie["id"])
     
     return (
-        trailer_url,
+        gr.HTML(value=trailer_html),
         selected_movie["title"],
         f"{selected_movie['title']}\n\nRating: {selected_movie['rating']}/10\n\n{selected_movie['overview']}"
     )
@@ -149,7 +132,11 @@ def rate_movie(movie_title: str, rating: float) -> str:
 
 # Gradio UI
 with gr.Blocks(theme=gr.themes.Soft()) as movie_app:
-    gr.Markdown("# 🎬 Movie Streaming App")
+    gr.Markdown("""
+    # 🎬 Movie Explorer
+    
+    Browse popular movies, watch trailers, and explore different genres!
+    """)
     
     with gr.Row():
         genre_dropdown = gr.Dropdown(
@@ -169,7 +156,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as movie_app:
         )
     
     with gr.Row():
-        video_player = gr.Video(label="Trailer")
+        trailer_html = gr.HTML(label="Trailer")
     
     with gr.Row():
         with gr.Column():
@@ -190,7 +177,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as movie_app:
     
     gallery.select(
         fn=get_movie_info,
-        outputs=[video_player, selected_movie_title, movie_info]
+        outputs=[trailer_html, selected_movie_title, movie_info]
     )
     
     rate_button.click(
@@ -199,6 +186,5 @@ with gr.Blocks(theme=gr.themes.Soft()) as movie_app:
         outputs=rating_output
     )
 
-# Launch app
 if __name__ == "__main__":
-    movie_app.launch(share=True)
+    movie_app.launch()