Update app.py

app.py

@@ -123,34 +123,67 @@ def generate_cad_from_text(text_prompt, max_length=512, temperature=0.8, top_p=0
     try:
         # Load model if not already loaded
         if not cad_fusion.model_loaded:
-            cad_fusion.load_model()
+            try:
+                cad_fusion.load_model()
+            except Exception as e:
+                error_msg = f"Failed to load CADFusion model: {str(e)}\n\nThis might be due to:\n- Model access restrictions\n- Insufficient resources\n- Network connectivity issues"
+                return error_msg, ""
+        
+        # Validate input
+        if not text_prompt or text_prompt.strip() == "":
+            return "Please provide a description for the CAD model.", ""
         
         # Generate CAD sequence
         cad_sequence = cad_fusion.generate_cad_sequence(
-            text_prompt, 
+            text_prompt.strip(), 
             max_length=int(max_length), 
             temperature=temperature, 
             top_p=top_p
         )
         
+        if not cad_sequence:
+            return "No CAD sequence was generated. Please try with a different prompt.", ""
+        
         # Create visualization info
         viz_info = cad_fusion.render_cad_visualization(cad_sequence)
         
-        # Format output
+        # Format detailed output
+        ops = viz_info.get('operations', {})
         output_text = f"""
+## 🎯 Generated CAD Model
+
+**Input Description:** {text_prompt}
+
 **Generated CAD Sequence:**
-{cad_sequence}
+```
+{cad_sequence[:500]}{'...' if len(cad_sequence) > 500 else ''}
+```
+
+## 📊 Analysis:
+- **Total Operations:** {viz_info.get('total_operations', 0)}
+- **Complexity:** {viz_info.get('complexity', 'Unknown')}
+- **Lines of Code:** {viz_info.get('line_count', 0)}
+
+### Operation Breakdown:
+- **Sketches:** {ops.get('sketch', 0)}
+- **Extrusions:** {ops.get('extrude', 0)}
+- **Revolutions:** {ops.get('revolve', 0)}
+- **Circles:** {ops.get('circle', 0)}
+- **Rectangles:** {ops.get('rectangle', 0)}
+- **Lines:** {ops.get('line', 0)}
+- **Fillets:** {ops.get('fillet', 0)}
+- **Chamfers:** {ops.get('chamfer', 0)}
+
+**Status:** {viz_info.get('status', 'Generated successfully')}
 
-**Analysis:**
-- Operations detected: {viz_info.get('operations', 0)}
-- Sketches detected: {viz_info.get('sketches', 0)}
-- Status: {viz_info.get('status', 'Generated')}
+---
+*Note: This is the parametric CAD sequence. For full 3D rendering, use CAD software that supports these operations.*
 """
         
         return output_text, cad_sequence
         
     except Exception as e:
-        error_msg = f"Error: {str(e)}"
+        error_msg = f"❌ Error generating CAD: {str(e)}"
         logger.error(error_msg)
         return error_msg, ""