import openai
import gradio as gr

# Set your OpenAI API key here
api_key = "YOUR_API_KEY"

# Initialize the OpenAI API client
openai.api_key = api_key

# Define a function to generate responses using GPT-3.5 Turbo
def generate_response(user_prompt):
    response = openai.Completion.create(
        model="gpt-3.5-turbo",  # Use GPT-3.5 Turbo engine
        prompt= f'''I will give you a question and you detect which category does this question belong to. It should be from these categories - 
        physical activity, sleep, nutrition and preventive care. Make sure you just reply with response in json format "category":"[sleep,nutrition]". 
        Note that single question may belong to multiple categories. Dont add any opening lines just reply with json response. 
        Question: {user_prompt}''',
        max_tokens=50,  # You can adjust this to limit the response length
    )
    return response.choices[0].text

# Create a Gradio interface
iface = gr.Interface(
    fn=generate_response,
    inputs="text",
    outputs="text",
    title="Detect Prompt Category",
    description="Enter a prompt, and GPT-3.5 Turbo will generate a response.",
)

# Start the Gradio interface
iface.launch()