Update app.py

app.py

@@ -7,20 +7,41 @@ classifier = pipeline("zero-shot-classification", model="cross-encoder/nli-disti
 # عنوان التطبيق
 st.title("Text Classification App")
 
-# إدخال النص
-text = st.text_area("Enter your text here:")
-
-# زر التصنيف
-if st.button("Classify"):
-    if text:
-        # تحديد الفئات التي تريد تصنيف النص إليها
-        categories = ["shopping", "gaming", "streaming"]
-        result = classifier(text, categories)
-        
-        # الحصول على الفئة الأكثر احتمالاً
+# إدخال الملف النصي
+uploaded_file = st.file_uploader("Upload a text file containing keywords", type=["txt"])
+
+if uploaded_file is not None:
+    # قراءة الملف النصي
+    content = uploaded_file.read().decode("utf-8")
+    keywords = [line.strip() for line in content.splitlines() if line.strip()]
+
+    # تحديد الفئات
+    categories = ["shopping", "gaming", "streaming"]
+
+    # قوائم لتخزين الكلمات حسب الفئة
+    shopping_words = []
+    gaming_words = []
+    streaming_words = []
+
+    # تصنيف الكلمات
+    for word in keywords:
+        result = classifier(word, categories)
         best_category = result['labels'][0]
-        score = round(result['scores'][0], 2)
-        
-        st.write(f"Classification Result: {best_category} (Confidence: {score})")
-    else:
-        st.warning("Please enter some text!")
+        if best_category == "shopping":
+            shopping_words.append(word)
+        elif best_category == "gaming":
+            gaming_words.append(word)
+        elif best_category == "streaming":
+            streaming_words.append(word)
+
+    # عرض النتائج في مربعات نصية قابلة للنسخ
+    st.header("Shopping Keywords")
+    st.text_area("Copy the shopping keywords here:", value="\n".join(shopping_words), height=200)
+
+    st.header("Gaming Keywords")
+    st.text_area("Copy the gaming keywords here:", value="\n".join(gaming_words), height=200)
+
+    st.header("Streaming Keywords")
+    st.text_area("Copy the streaming keywords here:", value="\n".join(streaming_words), height=200)
+else:
+    st.warning("Please upload a text file to classify the keywords.")