Create app.py

app.py

@@ -0,0 +1,55 @@
+import openai
+import gradio as gr
+import random
+
+def generate_dialogues(api_key, generate_button):
+    # Ensure the API key is set
+    openai.api_key = api_key
+
+    # Sample prompts provided
+    sample_prompts = [
+        "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!"
+    ]
+
+    # Randomly select one of the provided prompts
+    selected_prompt = random.choice(sample_prompts)
+
+    # Build the prompt to guide the model towards generating a one-line dialogue
+    prompt_text = (
+        f"Based on the scenario: '{selected_prompt}', generate a one-line dialogue:\n"
+    )
+
+    # Generate dialogue based on the selected prompt
+    response = openai.Completion.create(
+        engine="davinci",
+        prompt=prompt_text,
+        max_tokens=18  # Adjusted to limit the response to a one-liner
+    )
+    dialogues = response['choices'][0]['text'].strip()
+
+    # Return the generated dialogues
+    return dialogues
+
+def run_interface():
+    # Define Gradio interface
+    iface = gr.Interface(
+        fn=generate_dialogues,
+        inputs=[
+            gr.Textbox(label="OpenAI API Key"),
+            gr.Button(label="Generate")
+        ],
+        outputs="text",
+        live=False  # Set to False so the function only runs when the button is pressed
+    )
+
+    # Launch the interface
+    iface.launch()
+
+if __name__ == "__main__":
+    run_interface()