Update app.py

app.py

@@ -1,92 +1,28 @@
 import gradio as gr
 
-# Movie recommendations by genre
+# Define a dictionary with movie genres and recommendations
 movie_recommendations = {
-    "Animated": {
-        "title": "Soul (2020)", 
-        "image_url": "https://image-url.com/soul.jpg"
-    },
-    "Fantasy": {
-        "title": "The Boy and the Heron (2023)", 
-        "image_url": "https://image-url.com/the-boy-and-the-heron.jpg"
-    },
-    "Superhero": {
-        "title": "Spider-Man: No Way Home (2021)", 
-        "image_url": "https://image-url.com/spider-man-no-way-home.jpg"
-    },
-    "Action": {
-        "title": "John Wick: Chapter 4 (2023)", 
-        "image_url": "https://image-url.com/john-wick-4.jpg"
-    },
-    "Adventure": {
-        "title": "Dune (2021)", 
-        "image_url": "https://image-url.com/dune.jpg"
-    },
-    "Drama": {
-        "title": "Nomadland (2020)", 
-        "image_url": "https://image-url.com/nomadland.jpg"
-    },
-    "Comedy": {
-        "title": "Free Guy (2021)", 
-        "image_url": "https://image-url.com/free-guy.jpg"
-    },
-    "Horror": {
-        "title": "A Quiet Place Part II (2021)", 
-        "image_url": "https://image-url.com/a-quiet-place-2.jpg"
-    },
-    "Thriller": {
-        "title": "Tenet (2020)", 
-        "image_url": "https://image-url.com/tenet.jpg"
-    },
-    "Sci-Fi": {
-        "title": "Blade Runner 2049 (2017)", 
-        "image_url": "https://image-url.com/blade-runner-2049.jpg"
-    },
-    "Romance": {
-        "title": "The Last Letter from Your Lover (2021)", 
-        "image_url": "https://image-url.com/the-last-letter.jpg"
-    },
-    "Mystery": {
-        "title": "Knives Out (2019)", 
-        "image_url": "https://image-url.com/knives-out.jpg"
-    },
-    "Musical": {
-        "title": "West Side Story (2021)", 
-        "image_url": "https://image-url.com/west-side-story.jpg"
-    },
-    "Biography": {
-        "title": "Bohemian Rhapsody (2018)", 
-        "image_url": "https://image-url.com/bohemian-rhapsody.jpg"
-    },
-    "Documentary": {
-        "title": "The Social Dilemma (2020)", 
-        "image_url": "https://image-url.com/social-dilemma.jpg"
-    },
-    "War": {
-        "title": "1917 (2019)", 
-        "image_url": "https://image-url.com/1917.jpg"
-    },
-    "Western": {
-        "title": "The Power of the Dog (2021)", 
-        "image_url": "https://image-url.com/power-of-the-dog.jpg"
-    }
+    "action": ["Mad Max: Fury Road", "John Wick", "Die Hard", "Spider-Man: No Way Home"],
+    "comedy": ["Superbad", "The Hangover", "Step Brothers"],
+    "drama": ["The Shawshank Redemption", "Forrest Gump", "The Godfather"],
+    "sci-fi": ["Inception", "The Matrix", "Interstellar"],
+    "horror": ["The Conjuring", "Get Out", "A Nightmare on Elm Street"],
+    "romance": ["Pride and Prejudice", "La La Land", "The Notebook"],
+    "animation": ["Toy Story", "Finding Nemo", "Shrek", "Inside Out", "The Boy and the Heron"]
 }
 
-# Function to recommend a movie based on the selected genre
+# Function to recommend a movie based on the genre
 def recommend_movie(genre):
-    movie = movie_recommendations.get(genre)
-    if movie:
-        return movie['title'], movie['image_url']
+    genre = genre.lower()
+    if genre in movie_recommendations:
+        return f"Here are some {genre.capitalize()} movie recommendations:\n" + "\n".join(movie_recommendations[genre])
     else:
-        return "No recommendation available", ""
+        return "Sorry, we don't have recommendations for that genre. Please try another genre."
 
-# Gradio Interface
-demo = gr.Interface(
-    fn=recommend_movie,
-    inputs=gr.inputs.Dropdown(choices=list(movie_recommendations.keys()), label="Select Genre"),
-    outputs=[gr.outputs.Textbox(label="Movie Title"), gr.outputs.Image(label="Movie Poster")],
-    title="Movie Recommendation"
-)
+# Create a Gradio interface
+demo = gr.Interface(fn=recommend_movie, inputs="text", outputs="text", title="Movie Recommendation System", description="Enter a movie genre to get recommendations.")
 
+# Launch the Gradio app
 demo.launch()
 
+