Update app.py

app.py

@@ -1,35 +1,28 @@
-import openai
 import gradio as gr
-import requests
 import random
-from io import StringIO
 from datasets import load_dataset
 
-# Load the dataset from Hugging Face Spaces
+# Load the dataset from Hugging Face
 dataset = load_dataset("Blessin/dialogues-one-liners")
 
-# Get the column name for the dialogue data
-column_name = dataset.column_names[0]
-
-# Access the dialogue data
-DIALOGUES = dataset["train"][column_name]
+# Extract the dialogues from the dataset
+DIALOGUES = dataset["train"]["dialogues"]
 
 def generate_statement():
-    # Generate a random statement from the dataset
-    index = random.randint(0, len(DIALOGUES) - 1)
-    return DIALOGUES[index]
+    """Return a random dialogue from the dataset."""
+    return random.choice(DIALOGUES)
 
 def main():
-    # Define the UI components using gr.Interface
+    # Define the UI using gr.Interface
     interface = gr.Interface(
-        fn=generate_statement,     # Function to call on button press
-        inputs=[],                 # No inputs required as we're just generating random dialogues
-        outputs="text",            # Output is a text area
-        live=True,                # Only generate statement after button press
-        description="Press the button to generate a statement from our dataset."
+        fn=generate_statement,  # Function to call on button press
+        inputs=[],              # No inputs required
+        outputs="text",         # Output is a text area
+        live=False,             # Only generate statement after button press
+        description="Press the button to generate a statement from the dataset."
     )
     
-    # Launch the UI on Spaces
+    # Launch the UI
     interface.launch(share=True)
 
 if __name__ == "__main__":