import gradio as gr
import os

UPLOAD_FOLDER = 'uploads'
if not os.path.exists(UPLOAD_FOLDER):
    os.makedirs(UPLOAD_FOLDER)

def upload_file(file):
    if file is None:
        return "No file uploaded"
    file_name = file.name
    file_path = os.path.join(UPLOAD_FOLDER, file_name)
    with open(file_path, "wb") as f:
        f.write(file.read())
    return f"File uploaded successfully: {file_name}"

def list_files():
    files = os.listdir(UPLOAD_FOLDER)
    return files

def delete_file(file_name):
    file_path = os.path.join(UPLOAD_FOLDER, file_name)
    if os.path.exists(file_path):
        os.remove(file_path)
        return f"File deleted successfully: {file_name}"
    else:
        return "File not found"

# Define Gradio interface
with gr.Blocks() as demo:
    with gr.Row():
        upload = gr.File(label="Upload File")
        upload_btn = gr.Button("Upload")
        upload_output = gr.Textbox(label="Upload Status")
        upload_btn.click(upload_file, inputs=upload, outputs=upload_output)
        
    with gr.Row():
        list_btn = gr.Button("List Files")
        list_output = gr.Textbox(label="Files List")
        list_btn.click(list_files, outputs=list_output)