Update app.py

app.py

@@ -6,28 +6,30 @@ import yaml
 from tools.final_answer import FinalAnswerTool
 from Gradio_UI import GradioUI
 
+import qrcode
+from smolagents.agent_types import AgentImage
+
 # Below is an example of a tool that does nothing. Amaze us with your creativity !
 @tool
-def generate_qr_code(text:str)-> str: #it's import to specify the return type
+def generate_qr_code(text:str)-> AgentImage: #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 generate a QR code image based on the input string.
+    """A tool that generates a QR code image and returns it.
     Args:
         text: The text or URL that needs to be encoded into the QR code.
     """
-    import qrcode
+
     try:
         qr = qrcode.QRCode(box_size=10, border=4)
         qr.add_data(text)
         qr.make(fit=True)
-        
         img = qr.make_image(fill_color="black", back_color="white").convert('RGB')
 
         save_path = "generated_qr.png"
         img.save(save_path)
-        from smolagents.agent_types import AgentImage
+        
         return AgentImage(save_path)
     except Exception as e:
-        return f"QR code generation failed: {str(e)}"
+        return f"Error generating QR code:{str(e)}"
     # return "What magic will you build ?"
 
 @tool