Upload app.py

app.py

@@ -89,12 +89,20 @@ try:
             'good': 5, 'bad': 6, 'yes': 7, 'no': 8, 'water': 9,
             'food': 10, 'home': 11, 'work': 12, 'school': 13, 'family': 14
         }
+        utils = None  # Set utils to None if import fails
     
     st.success("✅ Core dependencies loaded successfully!")
 except ImportError as e:
     st.error(f"❌ Critical import error: {e}")
     st.error("Running in fallback mode with limited functionality.")
 
+# Import pose_utils separately to ensure it's always available
+try:
+    import pose_utils as utils
+except ImportError as utils_error:
+    st.error(f"❌ Failed to import pose_utils: {utils_error}")
+    utils = None
+
 # System information will be shown in About section
 
 
@@ -727,28 +735,34 @@ elif app_mode == 'Test Video Translation':
                     
                     if len(frame_buffer) < window_size:
                         # Initial frames - build up buffer
-                        visualization_canvas = utils.render_stick_model(
-                            frame,
-                            eval(frame_data['bodypose_circles']),
-                            eval(frame_data['bodypose_sticks']),
-                            eval(frame_data['handpose_edges']),
-                            eval(frame_data['handpose_peaks'])
-                        )
+                        if utils is not None:
+                            visualization_canvas = utils.render_stick_model(
+                                frame,
+                                eval(frame_data['bodypose_circles']),
+                                eval(frame_data['bodypose_sticks']),
+                                eval(frame_data['handpose_edges']),
+                                eval(frame_data['handpose_peaks'])
+                            )
+                        else:
+                            visualization_canvas = frame  # Use original frame if utils not available
                         
                         # Add prediction plots
-                        canvas_with_predictions = utils.create_bar_plot_visualization(
-                            visualization_canvas, {}, 
-                            f'Building Buffer - Frame {frame_idx + 1} [No Predictions Yet]',
-                            visualization_canvas
-                        )
-                        canvas_with_predictions = utils.create_bar_plot_visualization(
-                            canvas_with_predictions, weighted_predictions,
-                            f'Weighted Average - Frame {frame_idx + 1} [No Predictions Yet]',
-                            visualization_canvas
-                        )
-                        canvas_with_predictions = utils.add_bottom_padding(
-                            canvas_with_predictions, (255, 255, 255), 100
-                        )
+                        if utils is not None:
+                            canvas_with_predictions = utils.create_bar_plot_visualization(
+                                visualization_canvas, {}, 
+                                f'Building Buffer - Frame {frame_idx + 1} [No Predictions Yet]',
+                                visualization_canvas
+                            )
+                            canvas_with_predictions = utils.create_bar_plot_visualization(
+                                canvas_with_predictions, weighted_predictions,
+                                f'Weighted Average - Frame {frame_idx + 1} [No Predictions Yet]',
+                                visualization_canvas
+                            )
+                            canvas_with_predictions = utils.add_bottom_padding(
+                                canvas_with_predictions, (255, 255, 255), 100
+                            )
+                        else:
+                            canvas_with_predictions = visualization_canvas  # Use base canvas if utils not available
                         
                         # Initialize video writer
                         if video_writer is None:
@@ -821,28 +835,34 @@ elif app_mode == 'Test Video Translation':
                         )
                         
                         # Create visualization
-                        visualization_canvas = utils.render_stick_model(
-                            frame,
-                            eval(frame_data['bodypose_circles']),
-                            eval(frame_data['bodypose_sticks']),
-                            eval(frame_data['handpose_edges']),
-                            eval(frame_data['handpose_peaks'])
-                        )
+                        if utils is not None:
+                            visualization_canvas = utils.render_stick_model(
+                                frame,
+                                eval(frame_data['bodypose_circles']),
+                                eval(frame_data['bodypose_sticks']),
+                                eval(frame_data['handpose_edges']),
+                                eval(frame_data['handpose_peaks'])
+                            )
+                        else:
+                            visualization_canvas = frame  # Use original frame if utils not available
                         
                         # Add prediction visualizations
-                        canvas_with_predictions = utils.create_bar_plot_visualization(
-                            visualization_canvas, current_predictions,
-                            f'Current Window Prediction (Frames {sequence_idx + 1}-{frame_idx + 1})',
-                            visualization_canvas
-                        )
-                        canvas_with_predictions = utils.create_bar_plot_visualization(
-                            canvas_with_predictions, weighted_predictions,
-                            f'Cumulative Weighted Average - Frame {frame_idx + 1}',
-                            visualization_canvas
-                        )
-                        canvas_with_predictions = utils.add_bottom_padding(
-                            canvas_with_predictions, (255, 255, 255), 100
-                        )
+                        if utils is not None:
+                            canvas_with_predictions = utils.create_bar_plot_visualization(
+                                visualization_canvas, current_predictions,
+                                f'Current Window Prediction (Frames {sequence_idx + 1}-{frame_idx + 1})',
+                                visualization_canvas
+                            )
+                            canvas_with_predictions = utils.create_bar_plot_visualization(
+                                canvas_with_predictions, weighted_predictions,
+                                f'Cumulative Weighted Average - Frame {frame_idx + 1}',
+                                visualization_canvas
+                            )
+                            canvas_with_predictions = utils.add_bottom_padding(
+                                canvas_with_predictions, (255, 255, 255), 100
+                            )
+                        else:
+                            canvas_with_predictions = visualization_canvas  # Use base canvas if utils not available
                         
                         video_writer.write_frame(canvas_with_predictions)