put plotly inside of a catch

reflex-dev · Aug 12, 2024 · 854d729 · 854d729
1 parent 4349d69
commit 854d729
Showing 1 changed file with 21 additions and 16 deletions.
diff --git a/reflex/ivars/base.py b/reflex/ivars/base.py
@@ -810,23 +810,28 @@ def create(
                 ),
             )
 
-        from plotly.graph_objects import Figure, layout
-        from plotly.io import to_json
-
-        if isinstance(value, Figure):
-            return LiteralObjectVar.create(
-                json.loads(str(to_json(value))), _var_type=Figure, _var_data=_var_data
-            )
+        try:
+            from plotly.graph_objects import Figure, layout
+            from plotly.io import to_json
+
+            if isinstance(value, Figure):
+                return LiteralObjectVar.create(
+                    json.loads(str(to_json(value))),
+                    _var_type=Figure,
+                    _var_data=_var_data,
+                )
 
-        if isinstance(value, layout.Template):
-            return LiteralObjectVar.create(
-                {
-                    "data": json.loads(str(to_json(value.data))),
-                    "layout": json.loads(str(to_json(value.layout))),
-                },
-                _var_type=layout.Template,
-                _var_data=_var_data,
-            )
+            if isinstance(value, layout.Template):
+                return LiteralObjectVar.create(
+                    {
+                        "data": json.loads(str(to_json(value.data))),
+                        "layout": json.loads(str(to_json(value.layout))),
+                    },
+                    _var_type=layout.Template,
+                    _var_data=_var_data,
+                )
+        except ImportError:
+            pass
 
         if isinstance(value, Base):
             return LiteralObjectVar.create(