Update app.py

app.py

@@ -51,20 +51,24 @@ def set_openai_key(api_key):
     openai.api_key = api_key
 
 def generate_statement(api_key, use_gpt):
-    # Set the OpenAI API key using the provided input
-    set_openai_key(api_key)
+    try:
+        # Set the OpenAI API key using the provided input
+        set_openai_key(api_key)
+
+        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)
 
-    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(