Update app.py

app.py

@@ -1,12 +1,6 @@
 import streamlit as st
 from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
 
-# Initialize session state
-if 'show_all' not in st.session_state:
-    st.session_state.show_all = False
-if 'results' not in st.session_state:
-    st.session_state.results = []
-
 id_to_cat = {0: 'Performance',
  1: 'Molecular Networks',
  2: 'Operating Systems',
@@ -197,6 +191,7 @@ st.title("Классификация статей 1")
 # Ввод данных
 title = st.text_input("Заголовок статьи")
 description = st.text_area("Краткое описание статьи", height=150)
+top = st.text_input("Top x%")
 
 # Кнопка "Классифицировать"
 if st.button("Классифицировать"):
@@ -204,28 +199,28 @@ if st.button("Классифицировать"):
         st.warning("Пожалуйста, заполните хотя бы одно поле.")
     else:
         with st.spinner("Идет классификация..."):  # Индикатор загрузки
-            st.session_state.results = classify_text(title, description)
-            st.session_state.show_all = False  # Reset to show only top 95%
-
-# Display results if available
-if st.session_state.results:
-    if st.session_state.show_all:
-        st.subheader("Полные результаты классификации:")
-        for label, score in st.session_state.results:
-            st.write(f"- **{label}**: {score:.4f}")
-    else:
-        st.subheader("Результаты классификации (top 95%):")
-        cumulative_prob = 0
-        for label, score in st.session_state.results:
-            st.write(f"- **{label}**: {score:.4f}")
-            cumulative_prob += score
-            if cumulative_prob >= 0.95:
-                break
-        
-        # Renamed button that refreshes the page
-        if st.button("Покажи все"):
-            st.session_state.show_all = True
-            st.experimental_rerun()  # Refresh the page
+            results = classify_text(title, description)
+            if results:
+              
+              cumulative_prob = 0
+              t = 0.95
+              t_str
+              if top:
+                  if (top >= 0) and (top =< 1):
+                      t = top
+                      t_str = str(top * 100)
+                  elif (top > 1) and (top =< 100):
+                      t = top / 100
+                      t_str = str(top)
+              
+              st.subheader(f'Результаты классификации (top {t * 100}%):')
+              for label, score in results:
+                  st.write(f"- **{label}**: {score:.4f}")
+                  cumulative_prob += score
+                  if cumulative_prob >= t:
+                      break
+            else:
+                st.info("Не удалось получить результаты классификации.")
 
-elif title or description:  # небольшой костыль, чтобы при старте не было предупреждения
-    st.warning("Пожалуйста, нажмите кнопку 'Классифицировать'.")
+elif title or description: #небольшой костыль, чтобы при старте не было предупреждения
+    st.warning("Пожалуйста, заполните хотя бы одно поле.")