Update app.py

app.py

@@ -1,100 +1,13 @@
 import gradio as gr
-import random
 
-def generate_poem(topic):
-  # A list of possible rhymes for each line
-  rhymes = [
-    ["sky", "fly", "high", "eye", "why"],
-    ["moon", "soon", "tune", "spoon", "dune"],
-    ["star", "far", "car", "bar", "jar"],
-    ["light", "night", "bright", "sight", "fight"],
-    ["day", "way", "say", "play", "stay"]
-  ]
-  # A list of possible adjectives for each line
-  adjectives = [
-    ["blue", "clear", "vast", "deep", "bright"],
-    ["white", "full", "round", "shiny", "silver"],
-    ["sparkling", "distant", "shooting", "glowing", "twinkling"],
-    ["warm", "soft", "gentle", "calm", "sweet"],
-    ["sunny", "happy", "beautiful", "lovely", "wonderful"]
-  ]
-  # A list of possible nouns for each line
-  nouns = [
-    ["sky", "heaven", "cloud", "bird", "plane"],
-    ["moon", "lune", "globe", "cheese", "dream"],
-    ["star", "fire", "light", "wish", "heart"],
-    ["light", "sun", "ray", "glow", "smile"],
-    ["day", "life", "joy", "love", "hope"]
-  ]
-  # A list of possible verbs for each line
-  verbs = [
-    ["see", "feel", "touch", "reach", "dream"],
-    ["watch", "follow", "sing", "dance", "love"],
-    ["catch", "hold", "keep", "make", "share"],
-    ["shine", "burn", "glow", "spark", "blaze"],
-    ["live", "enjoy", "celebrate", "cherish", "embrace"]
-  ]
-  # A list of possible prepositions for each line
-  prepositions = [
-    ["in", "on", "under", "above", "over"],
-    ["by", "with", "from", "to", "for"],
-    ["of", "about", "around", "through", "across"],
-    ["like", "as", "than", "more", "less"],
-    ["at", "during", "since", "until", "after"]
-  ]
-  # A list of possible connectors for each line
-  connectors = [
-    ["and", "but", "or", "nor", "yet"],
-    ["so", "for", "because", "although", "though"],
-    ["if", "then", "else", "when", "while"],
-    ["where", "how", "why", "what", "who"],
-    ["which", "that", "this", "these", "those"]
-  ]
-  # A template for each line of the poem
-  template = [
-    "I {verb} the {adjective} {noun} {preposition} the {rhyme}",
-    "{connector}, I {verb} the {adjective} {noun} {preposition} the {rhyme}",
-    "I {verb} the {adjective} {noun} {preposition} the {rhyme}",
-    "{connector}, I {verb} the {adjective} {noun} {preposition} the {rhyme}",
-    "I {verb} the {adjective} {noun} {preposition} the {rhyme}"
-  ]
-  # Choose a random rhyme scheme for the poem
-  rhyme_scheme = random.choice(rhymes)
-  # Generate the poem line by line
-  poem = ""
-  for i in range(5):
-    # Choose a random word for each placeholder in the template
-    verb = random.choice(verbs[i])
-    adjective = random.choice(adjectives[i])
-    noun = random.choice(nouns[i])
-    preposition = random.choice(prepositions[i])
-    connector = random.choice(connectors[i])
-    rhyme = rhyme_scheme[i]
-    # Replace the placeholders with the chosen words
-    line = template[i].format(verb=verb, adjective=adjective, noun=noun, preposition=preposition, connector=connector, rhyme=rhyme)
-    # Capitalize the first letter of the line
-    line = line[0].upper() + line[1:]
-    # Add the line to the poem
-    poem += line
-    # Add a comma or a period at the end of the line
-    if i < 4:
-      poem += ","
-    else:
-      poem += "."
-    # Add a newline character after the line
-    poem += "\n"
-    # Yield the poem so far
-    yield poem
 
-# Create a text streaming mode interface with gradio
-demo = gr.Interface(
-  generate_poem, # The function to run
-  gr.Textbox("Enter a topic for the poem"), # The input component
-  gr.Textbox(streaming=True), # The output component
-  title="Poem Generator", # The title of the interface
-  description="This is a simple poem generator that creates a 5-line poem based on a topic. The poem has a random rhyme scheme and uses random words from a predefined list. Enter a topic and see the poem being generated in real time." # The description of the interface
-)
+def my_streaming_function(input_text):
+    for i in range(5):
+        yield f"Processing step {i+1}: {input_text}"
+        time.sleep(1)  # Simulate processing time
 
-# Launch the interface
-demo.launch()
-    
+gr.Interface(
+    my_streaming_function,
+    gr.Textbox(lines=2, label="Input text"),
+    gr.Textbox(streaming=True, label="Streaming output"),
+)