Update app.py

app.py

@@ -9,45 +9,53 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 
-OMDB_API_KEY = "OMDB_API_KEY"
-
 @tool
-def get_movies_by_mood(mood: str) -> str:
+def suggest_top_5_movies_by_mood(mood: str) -> list:
     """
-    Suggests up to 5 movies based on user's mood using OMDb API.
-
+    Suggest the top 5 movies based on the user's mood using DuckDuckGoSearchTool.
+    
     Args:
-        mood: Current emotional status or feeling of a user (e.g., 'sad', 'lonely', 'romantic').
+        mood: The user's current mood (e.g., "happy", "sad", "romantic", "bored").
+    
+    Returns:
+        list: A list of dictionaries, each containing 'title', 'description', and 'imdb_link'.
     """
-    mood_keywords = {
-        "happy": ["comedy", "feel good"],
-        "sad": ["drama", "emotional"],
-        "romantic": ["romance", "love"],
-        "inspired": ["biography", "true story"],
-        "bored": ["action", "thriller"],
-        "stressed": ["family", "light"],
-        "excited": ["adventure", "epic"],
-        "lonely": ["friendship", "self discovery"]
-    }
-
-    fallback_keywords = mood_keywords.get(
-        next((k for k in mood_keywords if k in mood.lower()), "sad"), ["drama"]
+    query = f"top 10 movies to watch when feeling {mood} site:imdb.com"
+    
+    temp_agent = CodeAgent(
+        tools=[DuckDuckGoSearchTool()],
+        model=model
     )
+    
+    search_results = temp_agent.run(query)
+    
+    # Use regex to extract IMDb links and nearby titles/descriptions
+    import re
+    pattern = r'https?://www\.imdb\.com/title/tt\d+[^ \n]*'
+    imdb_links = re.findall(pattern, search_results)
+    
+    lines = [line.strip() for line in search_results.split('\n') if line.strip()]
+    movies = []
+
+    for i, link in enumerate(imdb_links[:5]):
+        # Try to find the line before or after the link as description
+        context = ""
+        for j, line in enumerate(lines):
+            if link in line:
+                # Look backward or forward for a potential title/description
+                context = lines[j-1] if j > 0 else lines[j+1]
+                break
+        
+        title_match = re.search(r'“([^”]+)”', context) or re.search(r'"([^"]+)"', context)
+        title = title_match.group(1) if title_match else context[:60]
+        
+        movies.append({
+            "title": title.strip(),
+            "description": context.strip(),
+            "imdb_link": link
+        })
 
-    for keyword in fallback_keywords:
-        res = requests.get(f"http://www.omdbapi.com/?s={keyword}&apikey={OMDB_API_KEY}&type=movie").json()
-        if "Search" in res:
-            movies = []
-            for movie in res["Search"][:5]:
-                info = requests.get(f"http://www.omdbapi.com/?i={movie['imdbID']}&apikey={OMDB_API_KEY}").json()
-                title = info.get("Title", "Unknown")
-                year = info.get("Year", "N/A")
-                plot = info.get("Plot", "No plot available.")
-                movies.append(f"{title} ({year}): {plot}")
-            return "\n\n".join(movies)
-
-    return "Sorry, I couldn't find any matching movies for that mood right now."
-
+    return movies
         
 @tool
 def get_current_time_in_timezone(timezone: str) -> str:
@@ -86,7 +94,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer, get_current_time_in_timezone, get_movies_by_mood], ## add your tools here (don't remove final answer)
+    tools=[final_answer, get_current_time_in_timezone, suggest_top_5_movies_by_mood, DuckDuckGoSearchTool], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,