changed display, edited readme

README.md

@@ -12,3 +12,14 @@ short_description: Private Project
 ---
 
 Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+
+
+
+
+imdb_top250.html - is a html save of the top 250 movies list on imdb.com
+imdb-scrape250.py - script to get list of movieID for all the 250 movies in the html (Saved in imdb_movie_codes.csv)
+
+database.py - script to populate a dataframe with all the important API fetch (saved in MovieDatabase.csv)
+
+
+

app.py

@@ -1,81 +1,91 @@
+import base64
 import gradio as gr
-import requests
 import pandas as pd
 
-#load csv
-try:
-    df = pd.read_csv('/Users/moizpro/Desktop/MoviesRecommender/MovieRecommender/imdb_movie_codes.csv')
-    print("df loaded..")
-except FileNotFoundError:
-    print(f"Error: CSV file not found at path")
-#access frist movieID
-try:
-    # Initialize current movie index
-    current_movie_index = 0
-    # Assign the first element
-    movie_id = df.iloc[current_movie_index, 0]
-    # Print the value
-    print(movie_id)
-except:
-    print("error fetching first element")    
+# Load the dataset
+data = pd.read_csv('MovieDatabase.csv')
 
-# Function to fetch movie details using the given movie ID
-def fetch_movie_info(movie_id):
-    api_key = "f2443b04"
-    url = f"http://www.omdbapi.com/?apikey={api_key}&i={movie_id}&plot=full"
-    response = requests.get(url)
-    
-    if response.status_code == 200:
-        movie_data = response.json()
-        if movie_data.get("Response") == "True":
-            title = movie_data.get("Title", "N/A")
-            year = movie_data.get("Year", "N/A")
-            plot = movie_data.get("Plot", "N/A")
-            imdb_rating = movie_data.get("imdbRating", "N/A")
-            box_office = movie_data.get("BoxOffice", "N/A")
-            genre = movie_data.get("Genre", "N/A")
-            
-            return title, year, genre, plot, imdb_rating, box_office
-        else:
-            return "Error", "Error", "Error", "Error", "Error", "Error"
-    else:
-        return f"Failed to fetch movie details. HTTP Status Code: {response.status_code}"
+# Initialize the current index
+current_index = [0]
 
-# Function to update movie details for the next movie
-def update_movie_details():    
-    global current_movie_index 
-    print(current_movie_index)
-    movie_id = df.iloc[current_movie_index, 0]
-    print(movie_id)
-    current_movie_index = (current_movie_index + 1) % len(df)  # Loop back to start
-    return fetch_movie_info(movie_id)
+def encode_image(image_path):
+    with open(image_path, "rb") as img_file:
+        return f"data:image/png;base64,{base64.b64encode(img_file.read()).decode('utf-8')}"
 
-# Gradio interface
-with gr.Blocks(css="styles.css") as app:
-    gr.Markdown("# Movie Information Display")
-    gr.Markdown("This app fetches and displays movie information from a list of IMDb movie IDs.")
-    
-    # Initial movie details (fetch the first movie)
-    initial_movie_details = update_movie_details()
-    
-    title_output = gr.Textbox(label="Title", lines=1, value=initial_movie_details[0])
-    year_output = gr.Textbox(label="Year", lines=1, value=initial_movie_details[1])
-    genre_output = gr.Textbox(label="Genre", lines=1, value=initial_movie_details[2])
-    plot_output = gr.Textbox(label="Plot", lines=5, value=initial_movie_details[3])
-    imdb_rating_output = gr.Textbox(label="IMDb Rating", lines=1, value=initial_movie_details[4])
-    box_office_output = gr.Textbox(label="Box Office", lines=1, value=initial_movie_details[5])
-    
-    # Slider for user rating (optional)
-    slider = gr.Slider(minimum=0, maximum=10, step=0.25, label="Rating Slider")
+# Encode images as base64
+imdb_logo = encode_image("assets/imdblogo.png")
+rotten_logo = encode_image("assets/rotten.png")
+metacritic_logo = encode_image("assets/metacritic.png")
+
+def display_movie(action):
+    # Update the index based on action
+    if action == "next" and current_index[0] < len(data) - 1:
+        current_index[0] += 1
+    elif action == "prev" and current_index[0] > 0:
+        current_index[0] -= 1
     
-    # Button to fetch next movie
-    next_button = gr.Button("Next")
+    # Extract movie details
+    movie = data.iloc[current_index[0]]
+    details = {
+        "title": f"# {movie['Title']}",
+        "poster_placeholder": "🎥 Movie Poster Placeholder 🎥",
+        "ratings": f"""
+        <div style="display: flex; gap: 15px; align-items: center;">
+            <div style="text-align: center;">
+                <img src="{imdb_logo}" alt="IMDb" style="width: 40px; height: auto;"/>
+                <p>{movie['IMDb']}</p>
+            </div>
+            <div style="text-align: center;">
+                <img src="{rotten_logo}" alt="Rotten Tomatoes" style="width: 40px; height: auto;"/>
+                <p>{movie['Rotten Tomatoes']}</p>
+            </div>
+            <div style="text-align: center;">
+                <img src="{metacritic_logo}" alt="Metascore" style="width: 40px; height: auto;"/>
+                <p>{movie['Metascore']}</p>
+            </div>
+        </div>
+        """,
+        "details": f"""
+        **Year:** {movie['Year']}  
+        **Rated:** {movie['Rated']}  
+        **Runtime:** {movie['Runtime']}  
+        **Genre:** {movie['Genre1']}, {movie['Genre2']}, {movie['Genre3']}  
+        **Director:** {movie['Director']}  
+        **Writer:** {movie['Writer']}  
+        **Plot:** {movie['Plot']}  
+        **Awards:** {movie['Awards']}  
+        **Box Office:** {movie['BoxOffice']}
+        """
+    }
+    return details["title"], details["poster_placeholder"], details["ratings"], details["details"]
+
+# Define Gradio interface
+with gr.Blocks() as app:
+    gr.Markdown("## 🎬 Movie Database Viewer 🎬")
     
-    # Action on button click
+    with gr.Row():
+        with gr.Column(scale=1):
+            poster = gr.Markdown("🎥 Movie Poster Placeholder 🎥", elem_id="poster")
+        with gr.Column(scale=2):
+            title = gr.Markdown("# Title Placeholder", elem_id="title")
+            ratings = gr.HTML("<div style='text-align: center;'>Ratings Placeholder</div>", elem_id="ratings")
+            movie_details = gr.Markdown("Details will appear here.", elem_id="details")
+
+    with gr.Row():
+        prev_button = gr.Button("⬅️ Previous")
+        next_button = gr.Button("Next ➡️")
+
+    # Interactivity for buttons
+    prev_button.click(
+        display_movie,
+        inputs=gr.Text(value="prev", visible=False),
+        outputs=[title, poster, ratings, movie_details]
+    )
     next_button.click(
-        fn=update_movie_details, 
-        inputs=[], 
-        outputs=[title_output, year_output, genre_output, plot_output, imdb_rating_output, box_office_output]
+        display_movie,
+        inputs=gr.Text(value="next", visible=False),
+        outputs=[title, poster, ratings, movie_details]
     )
 
+# Launch the app
 app.launch()