Update app.py

Changed output type

app.py

@@ -10,35 +10,42 @@ from Gradio_UI import GradioUI
 # A tool that fetches data from an API endpoint
 import qrcode
 import os
+from PIL import Image
+import io
+import base64
 
 @tool
 def generate_qr_code(data: str) -> str:
-    """Generates a QR code from the given data and saves it as an image.
+    """Generates a QR code from the given data and returns it as a base64 string.
     Args:
         data: The text or URL to encode in the QR code.
     Returns:
-        Filepath to the generated QR code image.
+        Base64 encoded QR code image.
     """
     img = qrcode.make(data)
-    img_path = "qrcode.png"
-    img.save(img_path)
-    return img_path
+    buffered = io.BytesIO()
+    img.save(buffered, format="PNG")
+    img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    return img_str
 
 @tool
 def generate_image_and_qr(prompt: str) -> dict:
-    """Generates an image from a text prompt and creates a QR code linking to the saved image.
+    """Generates an image from a text prompt and returns both the image and QR code as base64 strings.
     Args:
         prompt: The text prompt for image generation.
     Returns:
-        A dictionary containing the image and QR code file paths.
+        A dictionary containing base64 encoded image and QR code.
     """
     try:
-        image_path = "generated_image.png"
         image = image_generation_tool(prompt=prompt)
-        image.save(image_path)
         
-        qr_code_path = generate_qr_code(image_path)
-        return {"image": image_path, "qr_code": qr_code_path}
+        buffered = io.BytesIO()
+        image.save(buffered, format="PNG")
+        image_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
+        
+        qr_code_str = generate_qr_code(image_str)
+        
+        return {"image": image_str, "qr_code": qr_code_str}
     except Exception as e:
         return {"error": str(e)}
 
@@ -59,9 +66,6 @@ def get_current_time_in_timezone(timezone: str) -> str:
 
 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' 
-
 model = HfApiModel(
     max_tokens=2096,
     temperature=0.5,
@@ -87,4 +91,4 @@ agent = CodeAgent(
     prompt_templates=prompt_templates
 )
 
-GradioUI(agent).launch()
+GradioUI(agent).launch()