| _C='API_KEY' | |
| _B='content' | |
| _A='role' | |
| from openai import OpenAI | |
| import gradio as gr,os | |
| API_KEY=os.environ[_C] | |
| client=OpenAI(base_url='https://openrouter.ai/api/v1',api_key=_C,default_headers={'Authorization':f"Bearer {API_KEY}",'HTTP-Referer':'','X-Title':''}) | |
| messages=[{_A:'system',_B:'Text Generation AI Assistant'}] | |
| def chatbot(input): | |
| if input:messages.append({_A:'user',_B:input});B=client.chat.completions.create(model='openai/gpt-oss-20b:free',messages=messages);A=B.choices[0].message.content;messages.append({_A:'assistant',_B:A});return A | |
| with gr.Interface(fn=chatbot,inputs=gr.Textbox(lines=3,label='Try: What is the value of pi',placeholder='Enter your message here...'),outputs=gr.Textbox(lines=10,label='Response'),title='',description='',theme=gr.themes.Default(primary_hue='sky'))as demo:demo.launch() |