Update app.py

app.py

@@ -4,32 +4,20 @@ import openai
 import gradio as gr
 
 def generate_improv_scene(api_key):
-    # Initialize the OpenAI API client with the provided key
-    openai.api_key = api_key
-
-    locations = ["beach", "outer space", "classroom", "pirate ship", "hospital", "concert stage", "underwater dome", "prehistoric cave", "unicorn land"]
-    emotions = ["happy", "angry", "confused", "elated", "sad", "anxious", "bemused", "bewildered"]
-    relationships = ["siblings", "co-workers", "friends", "enemies", "strangers", "parent and child", "alien and astronaut", "wizard and muggle"]
-    situations = ["finding a hidden portal", "preparing for an intergalactic festival", "solving the riddle of a talking frog", "escaping from a time loop", "organizing a dragon race", "trying to sell sand in a desert", "teaching dance to robots"]
-    occupations = ["astronaut", "dragon tamer", "unicorn whisperer", "intergalactic chef", "time traveler", "wizard", "robotic engineer", "underwater archaeologist"]
-
-    location, situation, occupation = random.sample(locations, 1)[0], random.sample(situations, 1)[0], random.sample(occupations, 1)[0]
-    emotion = random.choice(emotions)
-    relationship = random.choice(relationships)
-    
-    # Generate a line/dialogue using OpenAI API
-    dialogue_prompt = "Create an unrelated and random line spoken by a character."
-    response = openai.Completion.create(engine="text-davinci-003", prompt=dialogue_prompt, max_tokens=150)
-    dialogue = response.choices[0].text.strip()
-
-    # Return the desired elements as a tuple
-    return dialogue, emotion, situation, location, relationship, occupation
+    # ... (rest of your code remains unchanged)
 
 # Create Gradio UI
 iface = gr.Interface(
     fn=generate_improv_scene,
     inputs=["text"],  # Input for OpenAI API Key
-    outputs=["text", "text", "text", "text", "text", "text"],
+    outputs=[
+        gr.outputs.Textbox(label="dialogue"),
+        gr.outputs.Textbox(label="emotion"),
+        gr.outputs.Textbox(label="situation"),
+        gr.outputs.Textbox(label="relationship"),
+        gr.outputs.Textbox(label="occupation"),
+        gr.outputs.Textbox(label="location")  # Assuming you also want to output location
+    ],
     live=True,
     title="Improv Scene Generator",
     description="Generate a random improv scene with dialogue, emotion, situation, location, relationship, and occupation."