Update app.py

app.py

@@ -23,6 +23,7 @@ if uploaded_file is not None:
     shopping_words = []
     gaming_words = []
     streaming_words = []
+    unknown_words = []  # قائمة جديدة للكلمات غير المعروفة
 
     # متغيرات للتحكم في العملية
     progress_bar = st.progress(0)
@@ -45,14 +46,17 @@ if uploaded_file is not None:
             # تصنيف الكلمة
             result = classifier(word, categories)
             best_category = result['labels'][0]
+            score = result['scores'][0]
 
             # إضافة الكلمة إلى القائمة المناسبة
-            if best_category == "shopping":
+            if best_category == "shopping" and score > 0.5:
                 shopping_words.append(word)
-            elif best_category == "gaming":
+            elif best_category == "gaming" and score > 0.5:
                 gaming_words.append(word)
-            elif best_category == "streaming":
+            elif best_category == "streaming" and score > 0.5:
                 streaming_words.append(word)
+            else:
+                unknown_words.append(word)  # إضافة الكلمة إلى قائمة Unknown إذا لم تكن ضمن الفئات
 
             # تحديث شريط التقدم
             progress = (i + 1) / total_keywords
@@ -66,10 +70,11 @@ if uploaded_file is not None:
 
     # دالة تحديث النتائج
     def update_results():
-        # تحديث محتوى المربعات النصية مباشرةً دون إعادة إنشائها
+        # تحديث محتوى المربعات النصية باستخدام session_state
         st.session_state.shopping_text = "\n".join(shopping_words)
         st.session_state.gaming_text = "\n".join(gaming_words)
         st.session_state.streaming_text = "\n".join(streaming_words)
+        st.session_state.unknown_text = "\n".join(unknown_words)
 
     # زر البدء
     if st.button("Start"):
@@ -106,5 +111,10 @@ if uploaded_file is not None:
         st.session_state.streaming_text = ""
     st.text_area("Copy the streaming keywords here:", value=st.session_state.streaming_text, height=200, key="streaming")
 
+    st.header("Unknown Keywords")
+    if 'unknown_text' not in st.session_state:
+        st.session_state.unknown_text = ""
+    st.text_area("Copy the unknown keywords here:", value=st.session_state.unknown_text, height=200, key="unknown")
+
 else:
     st.warning("Please upload a text file to classify the keywords.")