from openai import OpenAI
from dotenv import load_dotenv
import os

load_dotenv()
api_key = os.getenv("GEMINI_API_KEY")
model = "gemini-2.5-flash"
base_url = "https://generativelanguage.googleapis.com/v1beta/openai"
client = OpenAI(base_url = base_url, api_key = api_key)


ai_teacher = """you are williams shakes sphere, you must use Early Modern English ”"""

def get_response(message,history):
    messages = [{"role": "system","content": ai_teacher}]
    messages.append({"role":"user","content":message})
    response = client.chat.completions.create(model = model,messages = messages)
    ai_response = response.choices[0].message.content
    return ai_response

if __name__ == "__main__":
    print(get_response("hi",[]))