Update app.py

app.py

@@ -1,55 +1,29 @@
 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
+# Set up the OpenAI API key
+openai.api_key = 'YOUR_OPENAI_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
+def generate_statement():
+    # Generate a statement using OpenAI's GPT 3.5 model
     response = openai.Completion.create(
         engine="davinci",
-        prompt=prompt_text,
-        max_tokens=18  # Adjusted to limit the response to a one-liner
+        prompt="Write a short one-line statement similar to common dialogue from movies or books:",
+        max_tokens=50
     )
-    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
+    return response.choices[0].text.strip()
+
+def main():
+    # Define the UI components using gr.Interface
+    interface = gr.Interface(
+        fn=generate_statement,  # Function to call on button press
+        inputs=[],              # No input components
+        outputs="text",         # Output is a text area
+        live=True               # Generate statement without requiring a button press
     )
-
-    # Launch the interface
-    iface.launch()
+    
+    # Launch the UI on Spaces
+    interface.launch(share=True)
 
 if __name__ == "__main__":
-    run_interface()
+    main()