Update app.py

app.py

@@ -417,14 +417,64 @@ await initHydra({{feedStrudel:5}})
     return complete_code
 
 # Main generation function
-def generate_interface(prompt, genre, complexity, include_visuals, visual_style, use_ai):
+def generate_interface(prompt, genre, complexity, include_visuals, 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."
     
-    # Always use curated pattern generation (removed AI for simplicity and speed)
-    strudel_code = generate_working_strudel_code(prompt, genre, complexity)
+    if use_ai:
+        # Use AI generation with better examples
+        system_prompt = f"""Generate working Strudel live coding pattern for: {prompt}
+Genre: {genre}, Complexity: {complexity}
+
+Example Strudel patterns:
+s("bd*4, hh*8").gain(0.8)
+stack(s("bd ~ sn ~"), n("0 2 4").s("sine").octave(3).gain(0.7))
+n("[0 3 5 7]*2").s("sawtooth").octave(2).lpf(1200).gain(0.8).scale("a:minor")
+
+Generate a {complexity} {genre} pattern:"""
+        
+        try:
+            outputs = code_generator(
+                system_prompt,
+                max_length=len(system_prompt.split()) + 80,
+                temperature=0.8,
+                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 more thoroughly
+            lines = strudel_code.split('\n')
+            clean_lines = []
+            for line in lines[:8]:  # Limit to 8 lines
+                line = line.strip()
+                # More robust pattern matching for Strudel code
+                if line and (line.startswith('s(') or line.startswith('n(') or line.startswith('stack(') or 
+                           '.gain(' in line or '.s(' in line or '.octave(' in line or '.scale(' in line or
+                           line.endswith(')') or line.endswith(',') or '.lpf(' in line):
+                    clean_lines.append(line)
+                # Stop at obvious non-Strudel patterns
+                elif any(stop_word in line.lower() for stop_word in ['function', 'var ', 'let ', 'const ', 'import', '//']):
+                    break
+            
+            if clean_lines:
+                strudel_code = '\n'.join(clean_lines)
+                strudel_code = f"// AI Generated: {prompt}\n{strudel_code}"
+            else:
+                # Fallback to curated patterns if AI fails
+                strudel_code = generate_working_strudel_code(prompt, genre, complexity)
+        except Exception as e:
+            # Fallback to curated patterns if AI fails
+            print(f"AI generation failed: {e}")
+            strudel_code = generate_working_strudel_code(prompt, genre, complexity)
+    else:
+        # Use curated pattern generation - ALWAYS RANDOM
+        strudel_code = generate_working_strudel_code(prompt, genre, complexity)
     
     # Create complete working template with random visuals based on genre
     complete_code = create_complete_strudel_code(strudel_code, include_visuals, genre)
@@ -473,6 +523,11 @@ with gr.Blocks(title="Working Strudel Generator", theme=gr.themes.Soft()) as app
                     label="🎨 Include visuals",
                     value=True
                 )
+                
+                use_ai = gr.Checkbox(
+                    label="🤖 Use AI (slower)",
+                    value=False
+                )
             
             generate_btn = gr.Button("Generate Code", variant="primary", size="lg")
         
@@ -522,8 +577,7 @@ with gr.Blocks(title="Working Strudel Generator", theme=gr.themes.Soft()) as app
             genre_dropdown,
             complexity_dropdown,
             include_visuals,
-            gr.State("random"),  # dummy visual_style
-            gr.State(False)      # dummy use_ai
+            use_ai
         ],
         outputs=output_code
     )