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)