Create app.py

app.py

@@ -0,0 +1,45 @@
+import gradio as gr
+import requests
+import json
+
+def get_code_screenshot(code, title, theme, background, darkMode, padding, language):
+    # Prepare the payload
+    payload = {
+        "code": code,
+        "title": title,
+        "theme": theme,
+        "background": background,
+        "darkMode": darkMode,
+        "padding": padding,
+        "language": language
+    }
+    # Send POST request to Rayso API
+    response = requests.post("https://rayso.herokuapp.com/api", json=payload)
+    # Check for a valid response
+    if response.status_code == 200:
+        # Convert response to json
+        data = response.json()
+        # Get image URL from response
+        image_url = data.get('url')
+        return image_url
+    else:
+        return f"Failed to get image: {response.status_code}"
+
+# Define Gradio interface
+iface = gr.Interface(
+    fn=get_code_screenshot,
+    inputs=[
+        gr.inputs.Textbox(lines=10, placeholder="Enter your code here..."),
+        gr.inputs.Textbox(default="Untitled-1", placeholder="Enter 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..."),
+    ],
+    outputs=[gr.outputs.Image(type="url")],
+    live=True
+)
+
+if __name__ == "__main__":
+    iface.launch()