import gradio as gr
from PIL import Image

import selinium


def get_image(name):
    selinium.main()
    return Image.open("reports/screenshot.png3_1.png")  # Replace with your image file

demo = gr.Interface(fn=get_image, inputs="text", outputs="image")
demo.launch()