import gradio as gr
from huggingface_hub import InferenceClient

models = {
    "Mistral-7B": "mistralai/Mistral-7B-Instruct-v0.2",
    "Falcon-7B": "tiiuae/falcon-7b-instruct"
}

def ask_model(question, model_choice):
    client = InferenceClient(models[model_choice])
    prompt = f"Answer this questionnaire item: {question} (Strongly disagree - Strongly agree)"
    return client.text_generation(prompt=prompt)

gr.Interface(
    fn=ask_model,
    inputs=[
        gr.Textbox(label="Questionnaire Item"),
        gr.Dropdown(list(models.keys()), label="Choose Model")
    ],
    outputs="text",
    title="AI Psychometrics: Data Collection"
).launch()