Update streamlit_app.py

streamlit_app.py

@@ -1,19 +1,8 @@
-import os, sys
-import google.generativeai as gen
-import inspect
-
-def env_check():
-    return {
-        "ver": getattr(gen, "__version__", None),
-        "file": getattr(gen, "__file__", None),
-        "has_responses": hasattr(gen, "responses"),
-        "has_GenerativeModel": hasattr(gen, "GenerativeModel"),
-        "has_generate": hasattr(gen, "generate"),
-        "python_version": sys.version,
-    }
-
-if __name__ == "__main__":
-    print(env_check())
+import streamlit as st
+from diagnostics import env_check
+
+st.header("Environment")
+st.code(env_check())
 
 # streamlit_app.py
 import os