nakas's picture
Update app.py
70a718e verified
raw
history blame
11.9 kB
import gradio as gr
import os
import json
import importlib
import sys
from contextlib import redirect_stdout, redirect_stderr
import io
# Set environment variable to avoid matplotlib permission issues
os.environ['MPLCONFIGDIR'] = '/tmp'
# Toggle for different applications
CURRENT_APP = "selector"
# Example app implementations
APP_REGISTRY = {
"hello_world": {
"title": "Hello World",
"description": "A simple greeting app",
"function": None # will be populated
},
"calculator": {
"title": "Calculator",
"description": "A basic arithmetic calculator",
"function": None # will be populated
},
"image_filter": {
"title": "Image Filter",
"description": "Apply visual effects to images",
"function": None # will be populated
}
}
# Hello World implementation
def create_hello_world():
def greet(name):
return f"Hello, {name}!"
demo = gr.Interface(
fn=greet,
inputs=gr.Textbox(label="Your Name"),
outputs=gr.Textbox(label="Greeting"),
title="Hello World App",
description="A simple greeting app",
examples=[["World"], ["Friend"], ["Gradio"]]
)
return demo
# Calculator implementation
def create_calculator():
def calculate(num1, num2, operation):
if operation == "Add":
return num1 + num2
elif operation == "Subtract":
return num1 - num2
elif operation == "Multiply":
return num1 * num2
elif operation == "Divide":
if num2 == 0:
return "Error: Division by zero"
return num1 / num2
demo = gr.Interface(
fn=calculate,
inputs=[
gr.Number(label="First Number"),
gr.Number(label="Second Number"),
gr.Radio(["Add", "Subtract", "Multiply", "Divide"], label="Operation")
],
outputs=gr.Textbox(label="Result"),
title="Calculator App",
description="Perform basic arithmetic operations",
examples=[
[5, 3, "Add"],
[10, 4, "Subtract"],
[6, 7, "Multiply"],
[20, 4, "Divide"]
]
)
return demo
# Image Filter implementation
def create_image_filter():
def apply_filter(image, filter_type):
if image is None:
return None
import numpy as np
from PIL import Image
img_array = np.array(image)
if filter_type == "Grayscale":
result = np.mean(img_array, axis=2).astype(np.uint8)
return Image.fromarray(result)
elif filter_type == "Invert":
result = 255 - img_array
return Image.fromarray(result)
elif filter_type == "Sepia":
sepia = np.array([[0.393, 0.769, 0.189],
[0.349, 0.686, 0.168],
[0.272, 0.534, 0.131]])
sepia_img = img_array.dot(sepia.T)
sepia_img[sepia_img > 255] = 255
return Image.fromarray(sepia_img.astype(np.uint8))
return image
demo = gr.Interface(
fn=apply_filter,
inputs=[
gr.Image(type="pil"),
gr.Radio(["Grayscale", "Invert", "Sepia"], label="Filter")
],
outputs=gr.Image(type="pil"),
title="Image Filter App",
description="Apply different filters to your images",
examples=[
["https://images.unsplash.com/photo-1507003211169-0a1dd7228f2d?w=500", "Grayscale"],
["https://images.unsplash.com/photo-1507003211169-0a1dd7228f2d?w=500", "Invert"],
["https://images.unsplash.com/photo-1507003211169-0a1dd7228f2d?w=500", "Sepia"]
]
)
return demo
# Register demo creation functions
APP_REGISTRY["hello_world"]["function"] = create_hello_world
APP_REGISTRY["calculator"]["function"] = create_calculator
APP_REGISTRY["image_filter"]["function"] = create_image_filter
# Example of code for each app as a string (for display purposes)
APP_CODES = {
"hello_world": """
import gradio as gr
def greet(name):
return f"Hello, {name}!"
demo = gr.Interface(
fn=greet,
inputs=gr.Textbox(label="Your Name"),
outputs=gr.Textbox(label="Greeting"),
title="Hello World App",
description="A simple greeting app",
examples=[["World"], ["Friend"], ["Gradio"]]
)
""",
"calculator": """
import gradio as gr
def calculate(num1, num2, operation):
if operation == "Add":
return num1 + num2
elif operation == "Subtract":
return num1 - num2
elif operation == "Multiply":
return num1 * num2
elif operation == "Divide":
if num2 == 0:
return "Error: Division by zero"
return num1 / num2
demo = gr.Interface(
fn=calculate,
inputs=[
gr.Number(label="First Number"),
gr.Number(label="Second Number"),
gr.Radio(["Add", "Subtract", "Multiply", "Divide"], label="Operation")
],
outputs=gr.Textbox(label="Result"),
title="Calculator App",
description="Perform basic arithmetic operations",
examples=[
[5, 3, "Add"],
[10, 4, "Subtract"],
[6, 7, "Multiply"],
[20, 4, "Divide"]
]
)
""",
"image_filter": """
import gradio as gr
import numpy as np
from PIL import Image
def apply_filter(image, filter_type):
if image is None:
return None
img_array = np.array(image)
if filter_type == "Grayscale":
result = np.mean(img_array, axis=2).astype(np.uint8)
return Image.fromarray(result)
elif filter_type == "Invert":
result = 255 - img_array
return Image.fromarray(result)
elif filter_type == "Sepia":
sepia = np.array([[0.393, 0.769, 0.189],
[0.349, 0.686, 0.168],
[0.272, 0.534, 0.131]])
sepia_img = img_array.dot(sepia.T)
sepia_img[sepia_img > 255] = 255
return Image.fromarray(sepia_img.astype(np.uint8))
return image
demo = gr.Interface(
fn=apply_filter,
inputs=[
gr.Image(type="pil"),
gr.Radio(["Grayscale", "Invert", "Sepia"], label="Filter")
],
outputs=gr.Image(type="pil"),
title="Image Filter App",
description="Apply different filters to your images",
examples=[
["https://images.unsplash.com/photo-1507003211169-0a1dd7228f2d?w=500", "Grayscale"],
["https://images.unsplash.com/photo-1507003211169-0a1dd7228f2d?w=500", "Invert"],
["https://images.unsplash.com/photo-1507003211169-0a1dd7228f2d?w=500", "Sepia"]
]
)
"""
}
# Function to create the selector app
def create_selector():
with gr.Blocks(title="Gradio App Generator") as demo:
gr.Markdown("# 🔄 Dynamic Gradio App Generator")
gr.Markdown("Select an app type to load it, or use the LLM to generate one")
with gr.Row():
with gr.Column(scale=1):
# App selection area
gr.Markdown("### 1. Choose an App Template")
app_selection = gr.Radio(
choices=list(APP_REGISTRY.keys()),
value="hello_world",
label="Available Templates",
info="Select a template to view and run it",
interactive=True
)
# Description display
app_description = gr.Markdown(f"**{APP_REGISTRY['hello_world']['title']}**: {APP_REGISTRY['hello_world']['description']}")
# Load button
load_btn = gr.Button("Load Selected App", variant="primary")
# Code display
with gr.Accordion("View Code", open=False):
code_display = gr.Code(
language="python",
value=APP_CODES["hello_world"],
label="App Code"
)
# Status message
status_msg = gr.Markdown("Select an app and click 'Load Selected App'")
# Simulated LLM area
gr.Markdown("### 2. Or Describe a New App")
app_description_input = gr.Textbox(
label="Describe the app you want",
placeholder="e.g., A calculator app that can add, subtract, multiply and divide",
lines=3
)
generate_btn = gr.Button("Generate & Load App")
with gr.Column(scale=1):
# Preview placeholder
preview_box = gr.Markdown(
"""
### App Preview
No app loaded yet. Select an app template or describe a new app to load.
"""
)
# Update description when selection changes
def update_description_and_code(selection):
return f"**{APP_REGISTRY[selection]['title']}**: {APP_REGISTRY[selection]['description']}", APP_CODES[selection]
app_selection.change(
update_description_and_code,
inputs=app_selection,
outputs=[app_description, code_display]
)
# Function to load the selected app
def load_app(app_name):
# Save the app name to a file so it persists
with open("current_app.txt", "w") as f:
f.write(app_name)
return f"Loading {APP_REGISTRY[app_name]['title']}... Refresh the page to see it."
load_btn.click(
load_app,
inputs=app_selection,
outputs=status_msg
)
# Function to handle app generation from description
def generate_app(description):
if not description:
return "Please enter a description first."
# Simple keyword matching to choose an app
description = description.lower()
if "hello" in description or "greet" in description:
app_name = "hello_world"
elif "calculat" in description or "math" in description or "add" in description:
app_name = "calculator"
elif "image" in description or "filter" in description or "photo" in description:
app_name = "image_filter"
else:
# Default to hello world
app_name = "hello_world"
# Save the app name to a file
with open("current_app.txt", "w") as f:
f.write(app_name)
return f"Generated and loading {APP_REGISTRY[app_name]['title']}... Refresh the page to see it."
generate_btn.click(
generate_app,
inputs=app_description_input,
outputs=status_msg
)
return demo
# Check which app to load
if os.path.exists("current_app.txt"):
try:
with open("current_app.txt", "r") as f:
app_name = f.read().strip()
if app_name in APP_REGISTRY:
CURRENT_APP = app_name
else:
CURRENT_APP = "selector"
except:
CURRENT_APP = "selector"
# Launch the appropriate app
if CURRENT_APP == "selector":
demo = create_selector()
else:
try:
# Try to create the requested app
demo = APP_REGISTRY[CURRENT_APP]["function"]()
except Exception as e:
# Fall back to selector on error
print(f"Error loading app {CURRENT_APP}: {str(e)}")
demo = create_selector()
# Launch the app
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=7860)