Update app.py

- integrated the OpenAI API to use GPT-3.5 for general conversation.
- added a new function `get_gpt_response()` that handles the interaction with the GPT-3.5 model.
- updated the `chatbot()` function to use GPT-3.5 for responses that aren't related to the quiz or card generation.
- added the `openai` library to the requirements.

app.py

@@ -2,6 +2,11 @@ import gradio as gr
 import random
 from diffusers import StableDiffusionPipeline
 import torch
+import openai
+import os
+
+# Set up OpenAI API
+openai.api_key = os.environ.get("OPENAI_API_KEY")
 
 # Quiz questions and answers
 christmas_quiz = [
@@ -41,9 +46,28 @@ def generate_image(prompt):
         image = pipe(prompt, num_inference_steps=50).images[0]
     return image
 
+def get_gpt_response(prompt, history):
+    messages = [
+        {"role": "system", "content": "You are a helpful Christmas-themed chatbot named Holly. You can answer questions about Christmas, offer holiday tips, and engage in festive conversation. You also know about the Christmas quiz and card generation features of this application."},
+    ]
+    for h in history:
+        messages.append({"role": "user", "content": h[0]})
+        messages.append({"role": "assistant", "content": h[1]})
+    messages.append({"role": "user", "content": prompt})
+    
+    response = openai.ChatCompletion.create(
+        model="gpt-3.5-turbo",
+        messages=messages,
+        max_tokens=150,
+        n=1,
+        stop=None,
+        temperature=0.7,
+    )
+    return response.choices[0].message['content'].strip()
+
 def chatbot(message, history):
     if not history:
-        return "Ho ho ho! Merry Christmas! I'm Santa's helper. Would you like to take a Christmas quiz or create a Christmas card?"
+        return "Ho ho ho! Merry Christmas! I'm Holly, your Christmas helper. Would you like to take a Christmas quiz, create a Christmas card, or chat about the holidays?"
 
     last_response = history[-1][1].lower()
 
@@ -60,17 +84,18 @@ def chatbot(message, history):
             if q['question'] in history[-2][1]:  # Find the question in the history
                 user_answer = q['options'][int(message) - 1]
                 if user_answer == q['answer']:
-                    return f"Correct! {q['answer']} is the right answer. Would you like another question or to create a Christmas card?"
+                    return f"Correct! {q['answer']} is the right answer. Would you like another question, to create a Christmas card, or to chat about something else?"
                 else:
-                    return f"Sorry, that's not correct. The right answer is {q['answer']}. Would you like another question or to create a Christmas card?"
+                    return f"Sorry, that's not correct. The right answer is {q['answer']}. Would you like another question, to create a Christmas card, or to chat about something else?"
 
     elif "card" in last_response:
         image = generate_image(f"Christmas card scene: {message}")
         return (f"I've created a Christmas card based on your description: '{message}'. You can see it in the image box below. "
-                f"Would you like to create another card or take a quiz?", image)
+                f"Would you like to create another card, take a quiz, or chat about something else?", image)
 
     else:
-        return "I'm not sure what you mean. Would you like to take a Christmas quiz or create a Christmas card?"
+        # Use GPT-3.5 for general conversation
+        return get_gpt_response(message, history)
 
 # Gradio interface
 with gr.Blocks() as demo:
@@ -79,7 +104,7 @@ with gr.Blocks() as demo:
     Welcome to the Christmas Quiz and Card Generator Chatbot!
     - Type 'quiz' to start a Christmas quiz.
     - Type 'card' to create a custom Christmas card.
-    - Or just say hello to get started!
+    - Or just chat about anything Christmas-related!
     """)
     chatbot = gr.Chatbot()
     msg = gr.Textbox(label="Type your message here")