import gradio as gr
import pandas as pd
import json
import uuid

def convert_inputs(user_input, var):
    if var == "__random__":
        return [f"session_{uuid.uuid4()}" for _ in user_input]

    if not isinstance(var, list):
        var = [var]
        
    var += var*(len(user_input)//len(var) + 1)

    diff = len(var) - len(user_input) 
    var = var[:-diff]
    
    return var

def process_inputs(user_input_json, session_id_json, project_id, chat_url, update_vars_json, output_vars_json, token):
    # Convert JSON strings into Python lists
    user_input = json.loads(user_input_json)
    session_id = json.loads(session_id_json)
    # project_id = json.loads(project_id_json)
    update_vars = json.loads(update_vars_json)
    output_vars = json.loads(output_vars_json)

    session_id = convert_inputs(user_input,session_id)
    # project_id = convert_inputs(user_input,project_id)
    update_vars = convert_inputs(user_input,update_vars)
    output_vars = convert_inputs(user_input,output_vars)

    # --- Your function logic here ---
    df = pd.DataFrame({
        "user_input": user_input,
        "session_id": session_id,
        "project_id": project_id,
        "chat_url": [chat_url] * len(user_input),
        "update_variables": update_vars,
        "output_variables": output_vars,
        "answer": [[] * len(x) if isinstance(x, list) else [] for x in user_input]
    })

    return df

def run_process(df):
    csv_path = "output.csv"
    df.to_csv(csv_path, index=False)
    return csv_path


with gr.Blocks() as demo:
    gr.Markdown("## 💬 JSON Input ➜ DataFrame ➜ CSV Export")

    user_input = gr.Code(label="user_input (list[str] or list[list[str]])", language="json", value='["Hello", ["Hi", "How are you?"]]')
    session_id = gr.Code(label="session_id (list[str] or str or '__random__')", language="json", value='["s1", "s2"]')
    
    # project_id = gr.Code(label="project_id (list[str])", language="json", value='["p1", "p2"]')
    
    project_id = gr.Textbox(label="Project ID", value="[project_id]")
    chat_url = gr.Textbox(label="CHAT URL", value="https://example.com/chat")
    token = gr.Textbox(label="TOKEN", value="[token]")
    
    update_vars = gr.Code(label="update_variables (list[dict])", language="json", value='["update1", "update2"]')
    output_vars = gr.Code(label="output_variables (list[str])", language="json", value='["out1", "out2"]')

    run_btn = gr.Button("Run Function")

    df_output = gr.Dataframe(label="Output DataFrame", interactive=True)
    
    process_btn = gr.Button("Process")
    
    file_output = gr.File(label="Download CSV")

    process_btn.click(
        fn=run_process,
        inputs=df_output,
        outputs=file_output
    )

    run_btn.click(
        fn=process_inputs,
        inputs=[user_input, session_id, project_id, chat_url, update_vars, output_vars, token],
        outputs=df_output
    )

demo.launch()