0xZohar commited on
Commit
5560e08
·
verified ·
1 Parent(s): daa3ea5

Fix: Add missing out_file parameter to render_bricks_safe() in CLIP flow

Browse files

- Fixed TypeError in CLIP text-to-ldr rendering (line 290)
- Added render_path creation before calling render_bricks_safe()
- Now matches original demo's calling pattern (line 399-402)
- Original file upload flow unchanged and still correct
- Bug was introduced when adding CLIP feature, not in original demo.zip

Files changed (1) hide show
  1. code/demo.py +4 -1
code/demo.py CHANGED
@@ -287,7 +287,10 @@ def unified_input_handler(file, text_query):
287
 
288
  # Render the GPT-generated LDR file
289
  print(f"🎨 Rendering GPT-generated LEGO design...")
290
- rendered_image = render_bricks_safe(new_ldr_path)
 
 
 
291
 
292
  # Update status message with generation info
293
  enhanced_status = f"✨ Generated from car_{car_id} (confidence: {confidence*100:.1f}%)\n🤖 GPT model created new assembly sequence\n{status}"
 
287
 
288
  # Render the GPT-generated LDR file
289
  print(f"🎨 Rendering GPT-generated LEGO design...")
290
+ render_filename = f"generated_{uuid.uuid4()}.png"
291
+ render_path = os.path.join(TMP_DIR, render_filename)
292
+ render_bricks_safe(new_ldr_path, render_path)
293
+ rendered_image = render_path
294
 
295
  # Update status message with generation info
296
  enhanced_status = f"✨ Generated from car_{car_id} (confidence: {confidence*100:.1f}%)\n🤖 GPT model created new assembly sequence\n{status}"