Update app.py

app.py

@@ -9,60 +9,39 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def moshaere(arg1: str) -> str:  # it's import to specify the return type
+def find_the_song(arg1: str) -> str:  # it's import to specify the return type
     # Keep this format for the description / args / args description but feel free to modify the tool
-    """A tool that continues the poetry by adding another verse to it. the verse should start with the ending alphabet of user verse
+    """A tool that finds the song name and description based on a user's given verse.
     Args:
         arg1: the user verse
         
     """
     try:
-        last_char = arg1[-1].lower()  # Get the last character (lowercase for consistency)
-        search_term = f"Persian poetry verse starting with {last_char}"  # Create a search query
+        search_term = f"song lyrics \"{arg1}\""  # Enclose the verse in quotes for exact match
         search_tool = DuckDuckGoSearchTool()  # Initialize the search tool
 
         search_results = search_tool.run(search_term)  # Perform the search
         # search_results = search_tool.use({"query": search_term}) #Old usage for smolagents, keep in case its needed.
 
-        # Check if the search results are not empty
+
         if search_results:
-            # Try to extract a relevant verse from the search results.
-            # This part needs careful handling to avoid hallucination.
-            # One approach is to look for text snippets containing " بیت " (verse) or other relevant keywords.
-            #Another way is just pick a random line from the results.
+            # Extract song information from the search results.
+            # This part requires careful extraction logic and may need improvement.
+            # Ideally, we want to find the song title, artist, and a brief description.
 
-            #Simple way of extracting verse (risky):
-            # verse = search_results.split('\n')[0] #get first line
             try:
-                verse_list = search_results.split("\n")
-                verse = None
-                for line in verse_list:
-                    if len(line)>2 and (line[0].isalpha()):
-                       verse = line
-                       break
-                if verse is None:
-                   return "Could not find an appropriate verse."
-
-
-                #Refine verse a little bit more:
-                refined_verse = verse
-                if ":" in refined_verse:
-                    refined_verse = refined_verse.split(":")[1]
-                if "-" in refined_verse:
-                    refined_verse = refined_verse.split("-")[1]
-
-                # Ensure the extracted verse starts with the correct letter:
-                if refined_verse[0].lower() == last_char:
-                    return refined_verse
-                else:
-                  return f"Could not find an appropriate verse directly from the search result starting with '{last_char}'. Please verify manually if there is a fitting one. \n Raw result: {search_results}"
-
-            except Exception as e:
-                return f"Error while extracting verse: {e} \n Raw Results: {search_results}"
+                # Simple approach:  Split by newline and assume the first few lines contain the song info.
+                lines = search_results.split("\n")
+                song_info = ""
+                for i in range(min(3, len(lines))):  # Consider the first 3 lines
+                    song_info += lines[i] + "\n"
 
+                return f"Possible song information:\n{song_info}\nRaw search result: {search_results}"
 
+            except Exception as e:
+                return f"Error extracting song info: {e}\nRaw search result: {search_results}"
         else:
-            return "Could not find any relevant verses."  # Handle empty search results
+            return "Could not find any songs matching the verse."  # Handle empty search results
 
     except Exception as e:
         return f"An error occurred: {e}"
@@ -104,7 +83,7 @@ with open("prompts.yaml", 'r') as stream:
     
 agent = CodeAgent(
     model=model,
-    tools=[final_answer], ## add your tools here (don't remove final answer)
+    tools=[final_answer, find_the_song,get_current_time_in_timezone], ## add your tools here (don't remove final answer)
     max_steps=6,
     verbosity_level=1,
     grammar=None,