Update app.py

app.py

@@ -33,9 +33,25 @@ def get_current_time_in_timezone(timezone: str) -> str:
     except Exception as e:
         return f"Error fetching time for timezone '{timezone}': {str(e)}"
 
+@def research_assistant(question: str) -> str:
+    ''' Ferramenta de pesquisa na web e geração de conteúdo.
+    Args: 
+    question: pergunta para a pesquisa
+    '''
+    search_results = DuckDuckGoSearchTool().run(question)
+
+    image_url = image_generation_tool(f"Infográfico sobre: {question[ :50]}...")
+
+    return f"""-> Resultados da pesquisa:
+{search_results}
+
+Visualização:
+{image_url}"""
+
 
 final_answer = FinalAnswerTool()
 
+
 # If the agent does not answer, the model is overloaded, please use another model or the following Hugging Face Endpoint that also contains qwen2.5 coder:
 # model_id='https://pflgm2locj2t89co.us-east-1.aws.endpoints.huggingface.cloud'