Update app.py

app.py

@@ -1,116 +1,68 @@
-document.addEventListener('DOMContentLoaded', function() {
-    // Poem data
-    const poems = [
-        {
-            text: "In the quiet of the evening, under skies so wide, I find my heart beating close to yours, my love. The stars above whisper secrets of our bond, eternal as the night, deep as the ocean's song.",
-            keywords: ["evening", "love", "stars"]
-        },
-        {
-            text: "Autumn leaves dance in crimson and gold, mirroring the fire that burns within my soul. Each leaf that falls is a promise made, that in your arms is where I'll fade.",
-            keywords: ["autumn", "leaves", "fire"]
-        },
-        {
-            text: "Your eyes are the moonlight that guides me through the dark, your touch the spark that ignites the fire in my heart. In this vast universe, you are my constant star, my love, my life, no matter near or far.",
-            keywords: ["moon", "eyes", "universe"]
-        },
-        {
-            text: "Like the river finds the sea, my heart has found its way to thee. Through storm and sunshine, come what may, in your love I'll always stay.",
-            keywords: ["river", "sea", "love"]
-        },
-        {
-            text: "In the garden of your heart, I plant my love so true. With every beat it grows, a flower just for you. The petals soft as whispers, the scent so sweet and light, blooming in the sunshine of your love so bright.",
-            keywords: ["garden", "heart", "flower"]
-        }
-    ];
+import gradio as gr
+import random
 
-    // DOM elements
-    const poemPrompt = document.getElementById('poem-prompt');
-    const generateBtn = document.getElementById('generate-btn');
-    const poemText = document.getElementById('poem-text');
-    const poemContainer = document.getElementById('poem-container');
-    const prevBgBtn = document.getElementById('prev-bg');
-    const nextBgBtn = document.getElementById('next-bg');
-    const bgSelect = document.getElementById('bg-select');
-
-    // Background images
-    const backgrounds = [
-        { id: 'autumn', path: 'assets/images/bg-autumn.svg' },
-        { id: 'stars', path: 'assets/images/bg-stars.svg' },
-        { id: 'hearts', path: 'assets/images/bg-hearts.svg' }
-    ];
-
-    let currentBgIndex = 0;
-
-    // Set initial background
-    updateBackground();
-
-    // Event listeners
-    generateBtn.addEventListener('click', generatePoem);
-    poemPrompt.addEventListener('keypress', function(e) {
-        if (e.key === 'Enter') {
-            generatePoem();
-        }
-    });
-
-    prevBgBtn.addEventListener('click', showPreviousBackground);
-    nextBgBtn.addEventListener('click', showNextBackground);
-    bgSelect.addEventListener('change', function() {
-        const selectedId = this.value;
-        const index = backgrounds.findIndex(bg => bg.id === selectedId);
-        if (index !== -1) {
-            currentBgIndex = index;
-            updateBackground();
-        }
-    });
-
-    // Functions
-    function generatePoem() {
-        const input = poemPrompt.value.toLowerCase();
-
-        // Simple filtering based on input keywords
-        let filteredPoems = poems;
-        if (input.trim()) {
-            const keywords = input.split(/\s+/);
-            filteredPoems = poems.filter(poem => {
-                return keywords.some(keyword =>
-                    poem.keywords.includes(keyword)
-                );
-            });
-        }
-
-        // If no matching poems, use all poems
-        if (filteredPoems.length === 0) {
-            filteredPoems = poems;
-        }
-
-        // Select random poem
-        const randomIndex = Math.floor(Math.random() * filteredPoems.length);
-        const selectedPoem = filteredPoems[randomIndex];
-
-        // Display poem with animation
-        poemText.style.opacity = 0;
-        setTimeout(() => {
-            poemText.textContent = selectedPoem.text;
-            poemText.style.opacity = 1;
-
-            // Change background
-            showNextBackground();
-        }, 300);
+# Sample data for backgrounds and poems
+backgrounds = ["autumn", "stars", "hearts"]
+poems = [
+    {
+        "text": "In the quiet of the evening, under skies so wide, I find my heart beating close to yours, my love. The stars above whisper secrets of our bond, eternal as the night, deep as the ocean's song.",
+        "keywords": ["evening", "love", "stars"]
+    },
+    {
+        "text": "Autumn leaves dance in crimson and gold, mirroring the fire that burns within my soul. Each leaf that falls is a promise made, that in your arms is where I'll fade.",
+        "keywords": ["autumn", "leaves", "fire"]
+    },
+    {
+        "text": "Your eyes are the moonlight that guides me through the dark, your touch the spark that ignites the fire in my heart. In this vast universe, you are my constant star, my love, my life, no matter near or far.",
+        "keywords": ["moon", "eyes", "universe"]
     }
+]
 
-    function updateBackground() {
-        const currentBg = backgrounds[currentBgIndex];
-        document.body.style.backgroundImage = `url('${currentBg.path}')`;
-        bgSelect.value = currentBg.id;
-    }
+def generate_poem(prompt):
+    # Simple filtering based on input keywords
+    filtered_poems = [poem for poem in poems if any(keyword.lower() in prompt.lower() for keyword in poem["keywords"])]
+    
+    # If no matching poems, use all poems
+    if not filtered_poems:
+        filtered_poems = poems
+    
+    # Select random poem
+    selected_poem = random.choice(filtered_poems)
+    return selected_poem["text"]
 
-    function showNextBackground() {
-        currentBgIndex = (currentBgIndex + 1) % backgrounds.length;
-        updateBackground();
-    }
+def change_background(direction):
+    global current_background_index
+    if direction == "next":
+        current_background_index = (current_background_index + 1) % len(backgrounds)
+    elif direction == "previous":
+        current_background_index = (current_background_index - 1) % len(backgrounds)
+    return backgrounds[current_background_index]
 
-    function showPreviousBackground() {
-        currentBgIndex = (currentBgIndex - 1 + backgrounds.length) % backgrounds.length;
-        updateBackground();
-    }
-});
+current_background_index = 0
+
+with gr.Blocks() as demo:
+    gr.Markdown("# Stellar Verse - Romantic Poetry Generator")
+    
+    with gr.Row():
+        poem_prompt = gr.Textbox(label="Enter themes or keywords for your poem", placeholder="Enter themes (love, moon, heart...)")
+        generate_btn = gr.Button("Generate Poem")
+        
+    with gr.Row():
+        prev_bg_btn = gr.Button("Previous Background", value="≺")
+        bg_select = gr.Dropdown(label="Select background", choices=backgrounds)
+        next_bg_btn = gr.Button("Next Background", value="≻")
+    
+    poem_text = gr.Textbox(label="Your romantic poem will appear here...", lines=5, interactive=False)
+    
+    def update_poem(prompt):
+        return generate_poem(prompt)
+    
+    def update_background(*args):
+        direction = args[0]
+        return change_background(direction)
+    
+    generate_btn.click(fn=update_poem, inputs=[poem_prompt], outputs=[poem_text])
+    prev_bg_btn.click(fn=update_background, inputs="previous", outputs=bg_select)
+    next_bg_btn.click(fn=update_background, inputs="next", outputs=bg_select)
+    
+demo.launch()