import subprocess
import sys

# Check if openai is installed, if not, install it
try:
    import openai
except ImportError:
    subprocess.check_call([sys.executable, "-m", "pip", "install", "openai"])
    import openai

import gradio as gr

# Replace with your actual OpenAI API key
openai.api_key = "your-api-key-here"

def get_python_help(message, history):
    # Construct the prompt
    prompt = f"""You are a helpful assistant that teaches Python programming. 
    You always refer to the latest Python documentation when answering questions. 
    If you're not sure about something, you say so and suggest checking the official documentation.
    
    User: {message}
    Assistant:"""

    # Get the full conversation history
    full_history = "\n".join([f"Human: {h[0]}\nAssistant: {h[1]}" for h in history])
    
    # Append the full history to the prompt
    full_prompt = f"{full_history}\n{prompt}"

    # Generate a response using GPT-3.5-turbo
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=[
            {"role": "system", "content": "You are a helpful assistant that teaches Python programming."},
            {"role": "user", "content": full_prompt}
        ],
        max_tokens=500,
        n=1,
        stop=None,
        temperature=0.7,
    )

    # Extract and return the generated message
    return response.choices[0].message['content'].strip()

# Create the Gradio interface
iface = gr.Interface(
    fn=get_python_help,
    inputs=["text", "state"],
    outputs=["text", "state"],
    title="Python Learning Assistant",
    description="Ask questions about Python programming and get help based on the latest documentation.",
    examples=[
        ["How do I create a list in Python?"],
        ["Explain Python decorators"],
        ["What's new in Python 3.10?"],
        ["How do I use f-strings?"],
        ["Explain the difference between tuples and lists"]
    ],
    theme="default"
)

# Launch the app
iface.launch()