Update Gradio_UI.py

Gradio_UI.py

@@ -30,7 +30,7 @@ from app import agent  # Импортируем объект агента
 
 def gradio_search_jokes(word):
     """Wrapper function that sends the request to the agent and retrieves the joke and its audio."""
-    response = agent.run(word)  # Отправляем запрос агенту
+    response = self.agent.run(word)  # Отправляем запрос агенту
     response_text = response.get("final_answer", "No response from agent.")  # Получаем текст ответа
 
     # Генерируем аудио
@@ -194,6 +194,7 @@ class GradioUI:
     """A one-line interface to launch your agent in Gradio"""
 
     def __init__(self, agent: MultiStepAgent, file_upload_folder: str | None = None):
+        self.agent = agent  # Сохраняем агента внутри класса
         if not _is_package_available("gradio"):
             raise ModuleNotFoundError(
                 "Please install 'gradio' extra to use the GradioUI: `pip install 'smolagents[gradio]'`"
@@ -296,7 +297,7 @@ class GradioUI:
             output_text = gr.Textbox(label="Jokes Found") #added
             output_audio = gr.Audio(label="Audio Pronunciation", type="filepath") #added
             btn = gr.Button("Get Jokes") #added
-            btn.click(gradio_search_jokes, inputs=input_box, outputs=[output_text, output_audio]) #added
+            btn.click(self.gradio_search_jokes, inputs=input_box, outputs=[output_text, output_audio]) #added