nakas's picture
Update app.py
2070667 verified
raw
history blame
13.7 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'
# 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"]
]
)
"""
}
# Create an app selector that loads apps dynamically
def create_main_app():
with gr.Blocks(title="Dynamic Gradio App Generator") as demo:
gr.Markdown("# 🔄 Dynamic Gradio App Generator")
gr.Markdown("Select an app type to load it dynamically without refreshing the page")
# Create placeholders for dynamic app loading
app_blocks_container = gr.Blocks(visible=False)
with app_blocks_container:
# This container will be dynamically replaced
app_placeholder = gr.Markdown("Loading app...")
with gr.Row():
with gr.Column(scale=1):
# App selection area
gr.Markdown("### 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'")
with gr.Column(scale=2):
# This is where our dynamically loaded app will appear
app_container = gr.Group(visible=True)
with app_container:
gr.Markdown("### App Preview")
gr.Markdown("No app loaded yet. Select an app and click 'Load Selected App'.")
# Update description and code 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]
)
# Load the selected app
@gr.on(load_btn, "click")
def load_selected_app(selection):
try:
# Create the app
app_func = APP_REGISTRY[selection]["function"]
app = app_func()
# Update the container content (this is done via JavaScript)
return {
status_msg: f"Successfully loaded {APP_REGISTRY[selection]['title']}",
app_container: gr.update(visible=False), # Hide temporarily for replacement
}
except Exception as e:
error_msg = str(e)
return {
status_msg: f"Error loading app: {error_msg}",
app_container: gr.update(visible=True),
}
# This uses Gradio's JavaScript event handling to dynamically mount the app
demo.load(None, js="""
function() {
let loadedAppIframe = null;
let intervalId = null;
// Function to handle the "Load Selected App" button click
document.addEventListener('click', function(event) {
// Find button by text or class
if (event.target.textContent === 'Load Selected App' ||
(event.target.parentElement && event.target.parentElement.textContent === 'Load Selected App')) {
// Get the selected app
const radioButtons = document.querySelectorAll('input[type="radio"]');
let selectedApp = null;
for (const radio of radioButtons) {
if (radio.checked) {
selectedApp = radio.value;
break;
}
}
if (!selectedApp) return;
// Stop any previous timer
if (intervalId) {
clearInterval(intervalId);
}
// Show loading indicator
const appContainer = document.querySelector('.app-preview');
if (appContainer) {
appContainer.innerHTML = '<p>Loading app...</p>';
}
// Create route for app
const appUrl = '/' + selectedApp;
// Clean up previous iframe if exists
if (loadedAppIframe) {
loadedAppIframe.remove();
}
// Create an iframe and load the app
loadedAppIframe = document.createElement('iframe');
loadedAppIframe.style.width = '100%';
loadedAppIframe.style.height = '600px';
loadedAppIframe.style.border = 'none';
loadedAppIframe.src = appUrl;
// Find the app container column (second column)
const columns = document.querySelectorAll('.gradio-column');
if (columns.length > 1) {
columns[1].innerHTML = ''; // Clear existing content
columns[1].appendChild(loadedAppIframe);
}
// Update the status
const statusEl = document.querySelector('#status-message');
if (statusEl) {
statusEl.textContent = 'App loaded successfully';
}
}
});
}
""")
return demo
# Function to create and mount all apps
def create_app():
# Create the main selector app
main_app = create_main_app()
# Create all the individual apps
for app_name, app_info in APP_REGISTRY.items():
app_func = app_info["function"]
if app_func:
app_instance = app_func()
# This is for Gradio 4.x - it mounts each app at its own route
main_app = gr.mount_gradio_app(main_app, app_instance, f"/{app_name}")
return main_app
# Launch the app
if __name__ == "__main__":
demo = create_app()
demo.launch(server_name="0.0.0.0", server_port=7860)