import gradio as gr
from transformers import pipeline

# Load YOUR custom model
philo_pipe = pipeline(
    "text-generation",
    model="mari-a-12a/opt-2.7b-philosophy",  # Replace with your model
    device="cpu"  # Switch to "cuda" if using GPU
)

def respond(message):
    full_prompt = f"Instruction: {message}\nOutput:"
    output = philo_pipe(full_prompt, max_length=100)[0]['generated_text']
    return output.split("AI:")[1].strip()

gr.Interface(
    fn=respond,
    inputs=gr.Textbox(lines=2, placeholder="Ask a philosophical question..."),
    outputs="text",
    title="Philosophy Chatbot (Fine-tuned OPT-2.7b)"
).launch(debug=True)