Upload streamlit_app.py with huggingface_hub

streamlit_app.py

@@ -67,17 +67,39 @@ if 'analysis_results' not in st.session_state:
 with st.sidebar:
     st.header("Configuration")
 
+    # Authentication method selection
+    auth_method = st.radio(
+        "Authentication Method",
+        ["Username/Password", "API Token"],
+        index=0
+    )
+
     # Zabbix API Configuration
     with st.expander("Zabbix API Settings", expanded=True):
         zabbix_url = st.text_input("Zabbix API URL", value=ZABBIX_API_URL)
-        zabbix_user = st.text_input("Username", value="Admin")
-        zabbix_password = st.text_input("Password", type="password")
+
+        if auth_method == "Username/Password":
+            zabbix_user = st.text_input("Username", value="Admin")
+            zabbix_password = st.text_input("Password", type="password")
+        else:
+            zabbix_token = st.text_input("API Token", type="password")
 
         if st.button("Connect to Zabbix"):
             with st.spinner("Connecting to Zabbix..."):
                 try:
                     zabbix = ZabbixAPI(zabbix_url)
-                    auth_token = zabbix.authenticate(zabbix_user, zabbix_password)
+
+                    if auth_method == "Username/Password":
+                        auth_token = zabbix.authenticate(zabbix_user, zabbix_password)
+                    else:
+                        # For API token authentication, we'll validate it by making a simple API call
+                        auth_token = zabbix_token
+                        # Test the token by making a simple API call
+                        test_result = zabbix.test_token_auth(auth_token)
+                        if not test_result:
+                            st.error("Invalid API token")
+                            auth_token = None
+
                     if auth_token:
                         st.session_state.zabbix_auth_token = auth_token
                         st.success("Successfully connected to Zabbix!")