Update app.py

app.py

@@ -51,23 +51,22 @@ def set_openai_key(api_key):
     openai.api_key = api_key
 
 def generate_statement(api_key, use_gpt):
-    try:
-        # Set the OpenAI API key using the provided input
-        set_openai_key(api_key)
+    # Set the OpenAI API key
+    set_openai_key(api_key)
+
+    if use_gpt:
+        # Generate a statement using OpenAI's davinci 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=15
+        )
+        return response.choices[0].text.strip()
+    else:
+        # Return a random dialogue from the provided list
+        return random.choice(DIALOGUES)
+
 
-        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)
-    except Exception as e:
-        return str(e)
 
 def main():
     # Define the UI components using gr.Interface