Update app.py

app.py

@@ -9,14 +9,63 @@ from Gradio_UI import GradioUI
 
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def my_custom_tool(arg1:str, arg2:int)-> 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 does nothing yet 
+def moshaere(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
     Args:
-        arg1: the first argument
-        arg2: the second argument
+        arg1: the user verse
+        
     """
-    return "What magic will you build ?"
+    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_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.
+
+            #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}"
+
+
+        else:
+            return "Could not find any relevant verses."  # Handle empty search results
+
+    except Exception as e:
+        return f"An error occurred: {e}"
 
 @tool
 def get_current_time_in_timezone(timezone: str) -> str: