Update test_app.py

test_app.py

@@ -1,16 +1,4 @@
-import pandas as pd
-import matplotlib
-import matplotlib.pyplot  as plt
-import plotly.express as px
-import numpy as np
-import plotly.graph_objects as go
-import sqlite3
-from typing import Optional, Dict, Any
-from datetime import datetime, timedelta
-import re
-from pathlib import Path
-
-
+import streamlit as st
 # import torch
 
 # # Give torch.classes a benign __path__ so Streamlit won't trigger __getattr__.
@@ -26,7 +14,29 @@ from pathlib import Path
 #             return getattr(self._obj, name)
 #     torch.classes = _TorchClassesWrapper(torch.classes)
 
-import streamlit as st
+
+
+if "modules_loaded" not in st.session_state:
+    # Do your big imports or setup here
+    # import torch
+    from predictor import EagleBlendPredictor
+    import pandas as pd
+    import matplotlib
+    import matplotlib.pyplot  as plt
+    import plotly.express as px
+    import numpy as np
+    import plotly.graph_objects as go
+    import sqlite3
+    from typing import Optional, Dict, Any
+    from datetime import datetime, timedelta
+    import re
+    from pathlib import Path
+    import time # Add this import to the top of your script
+    import math
+    import plotly.graph_objects as go
+    
+    st.session_state["modules_loaded"] = True
+
 
 ##---- fucntions ------
 # Load fuel data from CSV (create this file if it doesn't exist)
@@ -622,8 +632,7 @@ def get_model(db_path="eagleblend.db"):
 #                                                       Dashboard Tab
 # ----------------------------------------------------------------------------------------------------------------------------------------------
 with tabs[0]:
-    import math
-    import plotly.graph_objects as go
+
 
     # NOTE: Assuming these functions are defined elsewhere in your application
     # from your_utils import get_model, get_activity_logs, get_blends_overview
@@ -1306,7 +1315,7 @@ with tabs[1]:
 #                                                   Optimization Engine Tab
 # ----------------------------------------------------------------------------------------------------------------------------------------------
 
-import time # Add this import to the top of your script
+
 
 # --- Add this new function to your functions section ---
 def dummy_optimization_function(targets, fixed_targets, components_data):