Update app.py

app.py

@@ -61,39 +61,47 @@ if uploaded_file is not None:
     file_content = uploaded_file.read().decode("utf-8")
     queries.extend([line.strip() for line in file_content.splitlines() if line.strip()])
 
-if st.button("Класифицирай"):
+# UI for button with spinner
+button_disabled = False
+if queries:
+    button_disabled = False
+else:
+    button_disabled = True
+
+if st.button("Класифицирай", disabled=button_disabled):
     if queries:
-        # Tokenize in batch
-        inputs = tokenizer(
-            queries,
-            return_tensors="pt",
-            truncation=True,
-            padding=True,
-            max_length=256
-        )
+        with st.spinner("Обработване..."):
+            # Tokenize in batch
+            inputs = tokenizer(
+                queries,
+                return_tensors="pt",
+                truncation=True,
+                padding=True,
+                max_length=256
+            )
 
-        # Run inference
-        with torch.no_grad():
-            outputs = model(**inputs)
+            # Run inference
+            with torch.no_grad():
+                outputs = model(**inputs)
 
-        logits = outputs.logits
-        predictions = logits.argmax(dim=-1).tolist()
-        probabilities = F.softmax(logits, dim=-1)
-        confidence_scores = probabilities.max(dim=-1).values.tolist()
+            logits = outputs.logits
+            predictions = logits.argmax(dim=-1).tolist()
+            probabilities = F.softmax(logits, dim=-1)
+            confidence_scores = probabilities.max(dim=-1).values.tolist()
 
-        # Използване на наличната label mapping от модела
-        id2label = model.config.id2label
+            # Използване на наличната label mapping от модела
+            id2label = model.config.id2label
 
-        results = []
-        for query, pred, conf in zip(queries, predictions, confidence_scores):
-            predicted_label = id2label.get(str(pred), id2label.get(pred, "Неизвестно"))
-            results.append({
-                "Заявка": query,
-                "Предсказано намерение": predicted_label,
-                "Доверие": f"{conf:.2f}"
-            })
+            results = []
+            for query, pred, conf in zip(queries, predictions, confidence_scores):
+                predicted_label = id2label.get(str(pred), id2label.get(pred, "Неизвестно"))
+                results.append({
+                    "Заявка": query,
+                    "Предсказано намерение": predicted_label,
+                    "Доверие": f"{conf:.2f}"
+                })
 
-        st.write("### Резултати:")
-        st.table(results)
+            st.write("### Резултати:")
+            st.dataframe(results, use_container_width=True)
     else:
         st.warning("Моля, въведете поне една заявка, преди да класифицирате.")