import gradio as gr
from transformers import pipeline

# Setting up the Coretex brain
# Using Qwen 2.5 3B - very powerful and free to use immediately
model_id = "Qwen/Qwen2.5-3B-Instruct"
pipe = pipeline("text-generation", model=model_id, device_map="auto")

def coretex_chat(user_input):
    messages = [{"role": "user", "content": user_input}]
    # Coretex thinks here...
    response = pipe(messages, max_new_tokens=256)
    return response[0]['generated_text'][-1]['content']

# The interface humans see (we'll add the API for your bot later)
demo = gr.Interface(
    fn=coretex_chat, 
    inputs=gr.Textbox(label="Message Coretex"), 
    outputs=gr.Textbox(label="Response"),
    title="Coretex AI System"
)

demo.launch()