Update src/app.py

src/app.py

@@ -1,18 +1,28 @@
-import os, types, streamlit as st
+import streamlit as st
+import os
+import types
 
-# Fetch the hidden code from env var
+# Retrieve your hidden code from the environment variable
 app_code = os.environ.get("APP_CODE", "")
+st.write(app_code)
 
-def execute_code(code_str):
-    module = types.ModuleType("dynamic_app")
+def execute_code(code_str: str):
+    """
+    Dynamically compile and execute the provided code string.
+    If it defines a `main()` function, call it.
+    """
     try:
-        exec(code_str, module.__dict__)
-        if hasattr(module, "main"):
-            module.main()
+        # Create a fresh module namespace
+        dynamic_module = types.ModuleType("dynamic_app")
+        exec(code_str, dynamic_module.__dict__)
+        # If the app defines a main(), invoke it
+        if hasattr(dynamic_module, "main"):
+            dynamic_module.main()
     except Exception as e:
-        st.error(f"Error in hidden code: {e}")
+        st.error(f"Error while executing hidden code: {e}")
 
+# Load and run, or show an error
 if app_code:
     execute_code(app_code)
 else:
-    st.error("APP_CODE is empty. Did you set it?")
+    st.error("Cannot load the app code. Did you set the APP_CODE secret correctly?")