simple version test interface

app.py

@@ -1,4 +1,65 @@
+# app.py
+
 import streamlit as st
+from transformers import pipeline
+import pandas as pd
+
+
+@st.cache(allow_output_mutation=True)
+def load_model():
+    return pipeline(model="seara/rubert-tiny2-ru-go-emotions")
+
+model = load_model()
+
+emotion_labels = {
+    "admiration": ("восхищение", "🌟"),
+    "amusement": ("веселье", "😄"),
+    "anger": ("злость", "😡"),
+    "annoyance": ("раздражение", "😠"),
+    "approval": ("одобрение", "👍"),
+    "caring": ("забота", "🤗"),
+    "confusion": ("непонимание", "😕"),
+    "curiosity": ("любопытство", "🧐"),
+    "desire": ("желание", "🔥"),
+    "disappointment": ("разочарование", "😞"),
+    "disapproval": ("неодобрение", "👎"),
+    "disgust": ("отвращение", "🤢"),
+    "embarrassment": ("смущение", "😳"),
+    "excitement": ("возбуждение", "🤩"),
+    "fear": ("страх", "😱"),
+    "gratitude": ("признательность", "🙏"),
+    "grief": ("горе", "😭"),
+    "joy": ("радость", "😊"),
+    "love": ("любовь", "❤️"),
+    "nervousness": ("нервозность", "😬"),
+    "optimism": ("оптимизм", "🌈"),
+    "pride": ("гордость", "🏅"),
+    "realization": ("осознание", "💡"),
+    "relief": ("облегчение", "😌"),
+    "remorse": ("раскаяние", "😔"),
+    "sadness": ("грусть", "😢"),
+    "surprise": ("удивление", "😲"),
+    "neutral": ("нейтральность", "😐")
+}
+
+st.title("🎭 Эмоциональный анализ текста")
+st.write("Введите текст на русском, и мы покажем, какие эмоции в нём распознаны.")
+
+text_input = st.text_area("Введите текст здесь:")
+
+if text_input:
+    results = model(text_input)
+
+    threshold = 0.1
+    filtered = [r for r in results if r["score"] >= threshold]
+    filtered.sort(key=lambda x: x["score"], reverse=True)
 
-x = st.slider('Select a value')
-st.write(x, 'squared is', x * x)
+    if filtered:
+        st.subheader("Распознанные эмоции:")
+        for item in filtered:
+            label = item["label"]
+            score = item["score"]
+            ru_label, emoji = emotion_labels.get(label, (label, "❓"))
+            st.markdown(f"**{ru_label.capitalize()}** {emoji} — {score:.2%}")
+    else:
+        st.info("Эмоции не распознаны с достаточной уверенностью.")

requirements.txt

@@ -0,0 +1,2 @@
+transformers
+torch