Update app.py

app.py

@@ -1,9 +1,100 @@
 import gradio as gr
+import random
 
-def text_streaming(input_text):
-    # Process the input text (you can replace this with your logic)
-    output_text = f"You entered: {input_text}"
-    return output_text
+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
 
-iface = gr.Interface(fn=text_streaming, inputs="text", outputs="text", live=True)
-iface.launch()
+# 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
+)
+
+# Launch the interface
+demo.launch()
+