Upload streamlit_app.py with huggingface_hub

streamlit_app.py

@@ -4,7 +4,7 @@ import json
 from datetime import datetime, timedelta
 import pandas as pd
 import time
-from utils import zabbix_api, ollama_analysis
+from utils import ZabbixAPI, OllamaAnalysis
 from config import ZABBIX_API_URL, OLLAMA_API_URL
 
 # Set page config
@@ -76,7 +76,8 @@ with st.sidebar:
         if st.button("Connect to Zabbix"):
             with st.spinner("Connecting to Zabbix..."):
                 try:
-                    auth_token = zabbix_api.authenticate(zabbix_url, zabbix_user, zabbix_password)
+                    zabbix = ZabbixAPI(zabbix_url)
+                    auth_token = zabbix.authenticate(zabbix_user, zabbix_password)
                     if auth_token:
                         st.session_state.zabbix_auth_token = auth_token
                         st.success("Successfully connected to Zabbix!")
@@ -89,13 +90,14 @@ with st.sidebar:
     with st.expander("Event Filters", expanded=True):
         if st.session_state.zabbix_auth_token:
             # Get host groups
-            host_groups = zabbix_api.get_host_groups(zabbix_url, st.session_state.zabbix_auth_token)
+            zabbix = ZabbixAPI(zabbix_url)
+            host_groups = zabbix.get_host_groups(st.session_state.zabbix_auth_token)
             selected_group = st.selectbox("Host Group", [""] + [g['name'] for g in host_groups])
 
             # Get hosts based on selected group
             hosts = []
             if selected_group:
-                hosts = zabbix_api.get_hosts_by_group(zabbix_url, st.session_state.zabbix_auth_token, selected_group)
+                hosts = zabbix.get_hosts_by_group(st.session_state.zabbix_auth_token, selected_group)
             selected_host = st.selectbox("Host", [""] + [h['name'] for h in hosts])
 
             # Time range
@@ -126,8 +128,7 @@ with st.sidebar:
                         end_time = int(time.time())
                         start_time = end_time - (time_range * 3600)
 
-                        events = zabbix_api.get_events(
-                            zabbix_url,
+                        events = zabbix.get_events(
                             st.session_state.zabbix_auth_token,
                             start_time,
                             end_time,
@@ -190,11 +191,8 @@ if st.session_state.events_data:
                 )
 
                 # Get analysis from Ollama
-                analysis = ollama_analysis.get_analysis(
-                    ollama_url,
-                    model_name,
-                    prompt
-                )
+                ollama = OllamaAnalysis(ollama_url)
+                analysis = ollama.get_analysis(model_name, prompt)
 
                 if analysis:
                     st.session_state.analysis_results = analysis