Update app.py

imports and debugging again

app.py

@@ -7,8 +7,13 @@ from tools.final_answer import FinalAnswerTool
 
 from Gradio_UI import GradioUI
 
+import requests
+from imdb import Cinemagoer
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
+
+
 def get_movie_recommendation(mood: str, location: str, user_preferences: dict) -> dict:
     """
     Recommend a movie based on the user's mood, current weather at the given location,
@@ -25,14 +30,12 @@ def get_movie_recommendation(mood: str, location: str, user_preferences: dict) -
     
     # Fetch weather data from a free API (wttr.in)
     weather_url = f"https://wttr.in/{location}?format=%C"
-    
     response = requests.get(weather_url)
+    
     if response.status_code != 200 or "Unknown location" in response.text:
         return {"error": f"Could not fetch weather data for {location}."}
-
     
-    
-    weather_condition = response.text.lower()
+    weather_condition = response.text.strip().lower()
     
     # Define mood & weather-based genre mapping
     genre_mapping = {
@@ -50,29 +53,36 @@ def get_movie_recommendation(mood: str, location: str, user_preferences: dict) -
         "thunderstorm": "Thriller"
     }
     
-    # Determine recommended genre
+    # Determine recommended genres
     recommended_genres = genre_mapping.get(mood, ["Drama"])
     for key in weather_mapping:
         if key in weather_condition:
             recommended_genres.append(weather_mapping[key])
     
-    # Fetch movie recommendations from a free API (OMDb alternative: Cinemagoer - IMDbPY)
-    from imdb import Cinemagoer
-
-        ia = Cinemagoer()
-        all_movies = ia.get_top50_movies_by_genre(recommended_genres[0])  # Fetch top 50 movies in that genre
-        
-        if not all_movies:
-            return {"error": "No suitable movie found."}
-        
-        best_movie = all_movies[0]  # Pick the top-rated movie in that genre
-        
-        return {
-            "title": best_movie.get("title", "Unknown"),
-            "genre": recommended_genres,
-            "description": best_movie.get("plot outline", "No description available."),
-            "rating": best_movie.get("rating", "No rating available."),
-        }
+    # Initialize IMDbPY
+    ia = Cinemagoer()
+    
+    # Fetch top movies from IMDb by genre
+    genre_movies = []
+    for genre in recommended_genres:
+        try:
+            top_movies = ia.get_top50_movies_by_genre(genre)
+            genre_movies.extend(top_movies)
+        except Exception:
+            continue  # Skip if genre is unavailable
+    
+    if not genre_movies:
+        return {"error": "No suitable movie found."}
+    
+    best_movie = genre_movies[0]  # Pick the first movie
+
+    return {
+        "title": best_movie.get("title", "Unknown"),
+        "genre": recommended_genres,
+        "description": best_movie.get("plot outline", "No description available."),
+        "rating": best_movie.get("rating", "No rating available."),
+    }
+