Update app.py

app.py

@@ -23,7 +23,10 @@ def initialize_agents():
     )
 
     # Agent 2: Perform a web search
-    find_lyrics_site_agent = ReactCodeAgent(tools=[VisitWebpageTool()], llm_engine=llm_engine)
+    find_lyrics_site_agent = ReactCodeAgent(
+        tools=[DuckDuckGoSearchTool()], 
+        llm_engine=llm_engine
+    )
     managed_find_lyrics_site_agent = ManagedAgent(
         agent=find_lyrics_site_agent,
         name="find_lyrics_site",
@@ -31,9 +34,11 @@ def initialize_agents():
     )
 
     # Agent 3: Extract lyrics from webpage
-    web_parsing_agent = ReactCodeAgent(tools=[VisitWebpageTool()], 
-                                       llm_engine=llm_engine,
-                                       additional_authorized_imports=['requests', 'bs4'])
+    web_parsing_agent = ReactCodeAgent(
+        tools=[VisitWebpageTool()], 
+        llm_engine=llm_engine,
+        additional_authorized_imports=['requests', 'bs4']
+    )
     managed_web_parsing_agent = ManagedAgent(
         agent=web_parsing_agent,
         name="web_parsing_lyrics",
@@ -65,26 +70,52 @@ def initialize_agents():
 # Streamlit App
 def main():
     st.title("Lyrics Finder App 🎵")
-    st.write("Enter the name of the song and optionally the artist to retrieve the lyrics.")
+    st.write("Enter the name of the song and optionally the artist to retrieve the lyrics, or use one of the example queries below.")
 
     authenticate_hf()
     manager_agent = initialize_agents()
 
-    # User input
-    user_query = st.text_input("Enter the song and/or artist (e.g., 'Shape of You by Ed Sheeran')", "")
+    # Default examples
+    examples = [
+        "Search for Pelo Suelto by Elsa y Elmar",
+        "Find the lyrics to Shape of You by Ed Sheeran",
+        "Get the lyrics for Hips Don't Lie by Shakira",
+    ]
+
+    selected_example = st.selectbox("Select an example query:", [""] + examples)
+    custom_query = st.text_input("Or enter your own song and artist (e.g., 'Shape of You by Ed Sheeran')", "")
+
+    # Determine the task
+    task = selected_example if selected_example else custom_query
 
     if st.button("Find Lyrics"):
-        if user_query.strip() == "":
-            st.error("Please enter a valid song or artist name!")
+        if task.strip() == "":
+            st.error("Please select an example or enter a valid song and artist name!")
         else:
-            try:
-                st.write("Processing your request, please wait...")
-                with st.spinner("Agents are working hard to find the lyrics..."):
-                    result = manager_agent.run(user_query)
-                st.success("Lyrics retrieved successfully!")
-                st.markdown(f"### Lyrics for your song\n{result}")
-            except Exception as e:
-                st.error(f"An error occurred: {e}")
+            st.write("Processing your request, please wait...")
+            with st.spinner("Agents are working hard to find the lyrics..."):
+
+                # Retry logic
+                max_attempts = 5
+                done = False
+                attempts = 0
+                final_lyrics = None
+
+                while not done and attempts < max_attempts:
+                    try:
+                        result = manager_agent.run(task)
+                        final_lyrics = result["Task outcome (extremely detailed version)"]
+                        done = True
+                    except Exception as e:
+                        st.warning(f"Attempt {attempts + 1}/{max_attempts} failed: {e}")
+                        attempts += 1
+
+                if final_lyrics:
+                    st.success("Lyrics retrieved successfully!")
+                    st.markdown(f"### Lyrics for your song\n{final_lyrics}")
+                else:
+                    st.error("Failed to retrieve lyrics after multiple attempts.")
 
 if __name__ == "__main__":
     main()
+