Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 from transformers import HfApiEngine
-from transformers.agents import ReactCodeAgent, ReactJsonAgent, HfApiEngine, ManagedAgent
+from transformers.agents import ReactCodeAgent, ReactJsonAgent, ManagedAgent
 from transformers.agents.search import DuckDuckGoSearchTool, VisitWebpageTool
 from huggingface_hub import login
 
@@ -62,33 +62,17 @@ 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, or click on one of the example queries below.")
+    st.write("Enter the name of the song and optionally the artist to retrieve the lyrics.")
 
     authenticate_hf()
     manager_agent = initialize_agents()
 
-    # Default examples
-    examples = [
-        "Search for Pelo Suelto by Elsa y Elmar",
-        "Get the lyrics for Hips Don't Lie",
-    ]
-
-    # Suggest examples as clickable buttons
-    st.markdown("### Example Queries")
-    example_task = None
-    for example in examples:
-        if st.button(example):
-            example_task = example
-
-    # Custom input for user queries
-    custom_query = st.text_input("Or enter your own song and artist (e.g., 'Shape of You by Ed Sheeran')", "")
-
-    # Determine the task
-    task = example_task if example_task else custom_query
+    # Input for user queries
+    custom_query = st.text_input("Enter your song and artist (e.g., 'Search for Pelo Suelto by Elsa y Elmar')", "")
 
     if st.button("Find Lyrics"):
-        if task.strip() == "":
-            st.error("Please select an example or enter a valid song and artist name!")
+        if custom_query.strip() == "":
+            st.error("Please enter a valid song and artist name!")
         else:
             st.write("Processing your request, please wait...")
             with st.spinner("Agents are working hard to find the lyrics..."):
@@ -101,7 +85,7 @@ def main():
 
                 while not done and attempts < max_attempts:
                     try:
-                        result = manager_agent.run(task)
+                        result = manager_agent.run(custom_query)
                         final_lyrics = result["Task outcome (extremely detailed version)"]
                         done = True
                     except Exception as e:
@@ -110,10 +94,20 @@ def main():
 
                 if final_lyrics:
                     st.success("Lyrics retrieved successfully!")
-                    st.markdown(f"### Lyrics for your song\n{final_lyrics}")
+
+                    # Display lyrics in a larger text area
+                    st.text_area("Lyrics Output", final_lyrics, height=400)
+
+                    # Button to download lyrics as a text file
+                    lyrics_file = f"{custom_query.replace(' ', '_')}_lyrics.txt"
+                    st.download_button(
+                        label="Download Lyrics as .txt",
+                        data=final_lyrics,
+                        file_name=lyrics_file,
+                        mime="text/plain"
+                    )
                 else:
                     st.error("Failed to retrieve lyrics after multiple attempts.")
 
 if __name__ == "__main__":
     main()
-