Update app.py

app.py

@@ -244,22 +244,62 @@ await initHydra({{feedStrudel:5}})
     if not generated_code.startswith("$:"):
         generated_code = f"$: {generated_code}"
     
-    complete_code = f"""{visual_code}{generated_code}
-
-// Global effects (uncomment to use)
-// all(x => x.fft(5).scope())"""
+    complete_code = f"""{visual_code}{generated_code}"""
     
     return complete_code
 
 # Main generation function
-def generate_interface(prompt, genre, complexity, include_visuals, visual_style):
+def generate_interface(prompt, genre, complexity, include_visuals, visual_style, use_ai):
     """Main interface function that generates working code"""
     
     if not prompt.strip():
         return "Please enter a description of the music you want to create."
     
-    # Generate working Strudel code
-    strudel_code = generate_working_strudel_code(prompt, genre, complexity)
+    if use_ai:
+        # Use AI generation
+        system_prompt = f"""Generate working Strudel live coding pattern for: {prompt}
+Genre: {genre}, Complexity: {complexity}
+
+Example patterns:
+s("bd*4").gain(0.8)
+stack(s("bd ~ ~ ~"), n("0 2 4").s("sine").octave(3))
+
+Pattern:"""
+        
+        try:
+            outputs = code_generator(
+                system_prompt,
+                max_length=len(system_prompt.split()) + 50,
+                temperature=0.7,
+                do_sample=True,
+                top_p=0.9,
+                num_return_sequences=1
+            )
+            
+            generated_text = outputs[0]['generated_text']
+            strudel_code = generated_text[len(system_prompt):].strip()
+            
+            # Clean the AI output
+            lines = strudel_code.split('\n')
+            clean_lines = []
+            for line in lines[:5]:  # Limit to 5 lines
+                line = line.strip()
+                if line and (line.startswith('s(') or line.startswith('n(') or line.startswith('stack(') or 
+                           '.gain(' in line or '.s(' in line or line.endswith(')')):
+                    clean_lines.append(line)
+            
+            if clean_lines:
+                strudel_code = '\n'.join(clean_lines)
+                strudel_code = f"// AI Generated: {prompt}\n{strudel_code}"
+            else:
+                # Fallback to random if AI fails
+                strudel_code = generate_working_strudel_code(prompt, genre, complexity)
+        except:
+            # Fallback to random if AI fails
+            strudel_code = generate_working_strudel_code(prompt, genre, complexity)
+    else:
+        # Use random generation
+        strudel_code = generate_working_strudel_code(prompt, genre, complexity)
     
     # Create complete working template
     complete_code = create_complete_strudel_code(strudel_code, include_visuals, visual_style)
@@ -269,7 +309,7 @@ def generate_interface(prompt, genre, complexity, include_visuals, visual_style)
 # Create Gradio interface
 with gr.Blocks(title="Working Strudel Generator", theme=gr.themes.Soft()) as app:
     gr.Markdown("""
-    # 🎵 Codel Strudel Code Generator
+    # 🎵 CODEL: Strudel Code Generator
     
     Generate Strudel live coding patterns! 
     
@@ -308,12 +348,17 @@ with gr.Blocks(title="Working Strudel Generator", theme=gr.themes.Soft()) as app
                     value=True
                 )
                 
-                visual_style = gr.Dropdown(
-                    choices=["reactive", "kaleidoscope", "flowing", "geometric"],
-                    value="reactive",
-                    label="👁️ Visual Style"
+                use_ai = gr.Checkbox(
+                    label="🤖 Use AI generation",
+                    value=False
                 )
             
+            visual_style = gr.Dropdown(
+                choices=["reactive", "kaleidoscope", "flowing", "geometric"],
+                value="reactive",
+                label="👁️ Visual Style"
+            )
+            
             generate_btn = gr.Button("🎵 Generate Working Code", variant="primary", size="lg")
         
         with gr.Column():
@@ -328,7 +373,6 @@ with gr.Blocks(title="Working Strudel Generator", theme=gr.themes.Soft()) as app
             1. **Copy** all the code above
             2. **Go to** [strudel.cc](https://strudel.cc)
             3. **Paste** and **click the play button** ▶️
-            4. **It will work!** 🎉
             
             **🔧 Customization:**
             - Change `.gain()` values (0.1 to 1.0)
@@ -363,7 +407,8 @@ with gr.Blocks(title="Working Strudel Generator", theme=gr.themes.Soft()) as app
             genre_dropdown,
             complexity_dropdown,
             include_visuals,
-            visual_style
+            visual_style,
+            use_ai
         ],
         outputs=output_code
     )