Update app.py

app.py

@@ -1,8 +1,11 @@
 import gradio as gr
 import requests
+from PIL import Image
+from io import BytesIO
 
 # Function to interact with the Rayso API
 def get_code_screenshot(code, title, theme, background, darkMode, padding, language):
+    # Constructing the payload to be sent to the Rayso API
     payload = {
         "code": code,
         "title": title,
@@ -12,28 +15,42 @@ def get_code_screenshot(code, title, theme, background, darkMode, padding, langu
         "padding": padding,
         "language": language
     }
+    # Sending a POST request to the Rayso API
     response = requests.post("https://rayso.herokuapp.com/api", json=payload)
+    # Checking the response status code
     if response.status_code == 200:
+        # Parsing the JSON response to obtain the image URL
         data = response.json()
         image_url = data.get('url')
-        return image_url
+        # Sending a GET request to fetch the image from the obtained URL
+        response = requests.get(image_url)
+        # Checking the response status code
+        if response.status_code == 200:
+            # Converting the image bytes to a PIL Image object using BytesIO
+            image = Image.open(BytesIO(response.content))
+            # Returning the PIL Image object to be displayed by Gradio
+            return image
+        else:
+            # Returning an error message if failed to fetch the image
+            return f"Failed to fetch image: {response.status_code}"
     else:
+        # Returning an error message if failed to get the image URL
         return f"Failed to get image: {response.status_code}"
 
 # Gradio Interface
 iface = gr.Interface(
-    fn=get_code_screenshot,
+    fn=get_code_screenshot,  # Function to be called on user input
     inputs=[
-        gr.inputs.Textbox(lines=10, placeholder="Enter your code here...", label="Code"),
-        gr.inputs.Textbox(default="Untitled-1", placeholder="Enter title...", label="Title"),
-        gr.inputs.Dropdown(choices=["breeze", "candy", "crimson", "falcon", "meadow", "midnight", "raindrop", "sunset"], label="Theme"),
-        gr.inputs.Checkbox(default=True, label="Background"),
-        gr.inputs.Checkbox(default=True, label="Dark Mode"),
-        gr.inputs.Dropdown(choices=["16", "32", "64", "128"], label="Padding"),
-        gr.inputs.Textbox(default="auto", placeholder="Enter language...", label="Language"),
+        gr.Textbox(lines=10, placeholder="Enter your code here...", label="Code"),
+        gr.Textbox(default="Untitled-1", placeholder="Enter title...", label="Title"),
+        gr.Dropdown(choices=["breeze", "candy", "crimson", "falcon", "meadow", "midnight", "raindrop", "sunset"], label="Theme"),
+        gr.Checkbox(default=True, label="Background"),
+        gr.Checkbox(default=True, label="Dark Mode"),
+        gr.Dropdown(choices=["16", "32", "64", "128"], label="Padding"),
+        gr.Textbox(default="auto", placeholder="Enter language...", label="Language"),
     ],
     outputs=[
-        gr.outputs.Image(type="url", label="Generated Image"),
+        gr.Image(label="Generated Image"),  # Updated to handle the PIL Image object returned by get_code_screenshot
     ],
     live=False  # Set to False to only call the function when the submit button is pressed
 )