Update app.py

app.py

@@ -1,30 +1,79 @@
 import openai
 import gradio as gr
+import random
+
+DIALOGUES = [
+    "Don’t look now, but they’re right behind you.",
+    "I need to talk to you about something.",
+    "Oh no! You’re hurt!",
+    "And that's why you're fired.",
+    "What’s that on your arm?",
+    "Is this seat taken?",
+    "I can’t believe you actually got me a pony!",
+    "Quick, pretend we've known each other for years!",
+    "You're not going to believe what I just heard.",
+    "How did you manage to get that stuck there?",
+    "The last time I saw that, things didn’t end well.",
+    "Wait, wasn’t that supposed to be a secret?",
+    "Why is everyone looking at us?",
+    "You promised you’d never bring that up again!",
+    "Guess what I found in the attic?",
+    "This isn't what it looks like, I swear!",
+    "That's the last place I'd expect to see you!",
+    "Did you just hear that noise too?",
+    "Remember when you dared me? Well, I did it.",
+    "Why is there a llama in the living room?",
+    "That's the exact opposite of what I told you to do!",
+    "Whose idea was it to bring the penguin?",
+    "You owe me big time for this one.",
+    "I thought you said you could juggle!",
+    "I've never seen anyone react to cake like that before.",
+    "That was supposed to be our little secret!",
+    "You won't believe who I bumped into at the market.",
+    "Where did all these chickens come from?",
+    "I might've accidentally set it to the wrong date.",
+    "I told you mixing those two was a bad idea!",
+    "You're holding it upside down, you know.",
+    "Why does that parrot keep calling your name?",
+    "You really thought that would impress me?",
+    "I never expected to see you in a tutu.",
+    "Someone's left a mysterious package at the door.",
+    "Oh, so that's where my missing sock went!",
+    "How do you always end up in these situations?",
+    "I've got some news you might not like.",
+    "Why do you have that map and shovel?",
+    "I've got a secret, but promise not to laugh!"
+]
+
 
 def set_openai_key(api_key):
     # Set up the OpenAI API key
     openai.api_key = api_key
 
-def generate_statement(api_key):
+def generate_statement(api_key, use_gpt=True):
     # Set the OpenAI API key using the provided input
     set_openai_key(api_key)
 
-    # Generate a statement using OpenAI's GPT 3.5 model
-    response = openai.Completion.create(
-        engine="davinci",
-        prompt="Provide a short and intriguing dialogue that can be incorporated into an improvisation scene to provoke unexpected reactions and drive the narrative in interesting directions.",
-        max_tokens=50
-    )
-    return response.choices[0].text.strip()
+    if use_gpt:
+        # Generate a statement using OpenAI's GPT 3.5 Turbo model
+        response = openai.Completion.create(
+            engine="gpt-3.5-turbo",
+            prompt="Provide a short and intriguing dialogue that can be incorporated into an improvisation scene to provoke unexpected reactions and drive the narrative in interesting directions.",
+            max_tokens=15
+        )
+        return response.choices[0].text.strip()
+    else:
+        # Return a random dialogue from the provided list
+        return random.choice(DIALOGUES)
 
 def main():
     # Define the UI components using gr.Interface
     interface = gr.Interface(
         fn=generate_statement,     # Function to call on button press
-        inputs="text",             # Input is a text field for API key
+        inputs=["text", "checkbox"],  # Input is a text field for API key and a checkbox for GPT option
         outputs="text",            # Output is a text area
         live=False,                # Only generate statement after button press
-        description="Enter your OpenAI API key and press the button to generate a statement."
+        description="Enter your OpenAI API key and select whether to use GPT or get a random statement, then press the button."
     )
     
     # Launch the UI on Spaces