app.py

import os
from dotenv import load_dotenv
from openai import OpenAI

# =========================
# LOAD ENV VARIABLES
# =========================
load_dotenv()

api_key = os.getenv("OPENAI_API_KEY")

if not api_key:
    raise ValueError("❌ API key not found. Please set OPENAI_API_KEY in .env")

# =========================
# INITIALIZE CLIENT
# =========================
client = OpenAI(api_key=api_key)

# =========================
# CORE FUNCTION
# =========================
def generate_response(user_input, temperature=0.7, max_tokens=200, system_message=None):
    if system_message is None:
        system_message = "You are a helpful AI assistant."

    try:
        response = client.chat.completions.create(
            model="gpt-5.3",
            messages=[
                {"role": "system", "content": system_message},
                {"role": "user", "content": user_input}
            ],
            temperature=temperature,
            max_tokens=max_tokens
        )

        return response.choices[0].message.content

    except Exception as e:
        return f"❌ Error: {str(e)}"


# =========================
# TEMPERATURE COMPARISON
# =========================
def compare_temperatures(prompt):
    temps = [0.2, 0.5, 0.9]

    print("\n🔍 Comparing Outputs:\n")

    for temp in temps:
        print(f"\n--- Temperature: {temp} ---")
        response = generate_response(prompt, temperature=temp)
        print(response)


# =========================
# INTERACTIVE CHAT
# =========================
def chat_mode():
    print("\n🤖 Chat Mode (type 'exit' to quit)\n")

    while True:
        user_input = input("You: ")

        if user_input.lower() == "exit":
            print("👋 Goodbye!")
            break

        temp = input("Set temperature (0.0–1.0, default=0.7): ")
        max_tok = input("Set max_tokens (default=200): ")

        temperature = float(temp) if temp else 0.7
        max_tokens = int(max_tok) if max_tok else 200

        response = generate_response(
            user_input,
            temperature=temperature,
            max_tokens=max_tokens
        )

        print(f"\nAI: {response}\n")


# =========================
# SYSTEM MESSAGE MODE
# =========================
def system_mode():
    print("\n🧠 Custom System Role Mode\n")

    system_message = input("Enter system behavior (e.g., 'You are a strict teacher'): ")
    user_input = input("Enter your prompt: ")

    response = generate_response(
        user_input,
        system_message=system_message
    )

    print(f"\nAI: {response}\n")


# =========================
# MAIN MENU
# =========================
def main():
    while True:
        print("\n===== OpenAI Python Playground =====")
        print("1. Chat Mode")
        print("2. Compare Temperatures")
        print("3. Custom System Role")
        print("4. Exit")

        choice = input("Select option: ")

        if choice == "1":
            chat_mode()

        elif choice == "2":
            prompt = input("Enter prompt: ")
            compare_temperatures(prompt)

        elif choice == "3":
            system_mode()

        elif choice == "4":
            print("👋 Exiting...")
            break

        else:
            print("❌ Invalid choice")


# =========================
# RUN APP
# =========================
if __name__ == "__main__":
    main()