Update Information_Retrieval.py

Information_Retrieval.py

@@ -13,7 +13,7 @@ from pyvi import ViTokenizer
 @st.cache_resource
 def open2list_vn(path):
         if path:
-            with open(path) as f:
+            with open(path, encoding="utf8") as f:
                 line = list(f.read().splitlines())
             return line
 def pre_progress(input):
@@ -60,31 +60,51 @@ def main():
     st.set_page_config(page_title="Information Retrieval", page_icon="📝")
 
     # giving a title to our page
-    st.title("Information Retrieval")
-    text = st.text_area(
-        "Please enter a text:",
-        placeholder="Enter your text here",
-        height=200,
-    )
+    col1, col2 = st.columns([2, 1])
+    col1.title("Information Retrieval")
+
+    col2.link_button("Explore my model", "https://huggingface.co/nguyennghia0902/textming_proj01_electra")
+
     question = st.text_area(
-        "Please enter a question:",
+        "QUESTION: Please enter a question:",
         placeholder="Enter your question here",
-        height=200,
+        height=15,
+    )
+    text = st.text_area(
+        "CONTEXT: Please enter a context:",
+        placeholder="Enter your context here",
+        height=100,
     )
 
     prediction = ""
 
+    upload_file = st.file_uploader("CONTEXT: Or upload a file with some contexts", type=["txt"])
+    if upload_file is not None:
+        text = upload_file.read().decode("utf-8")
+
+        for line in text.splitlines():
+            line = line.strip()
+            if not line:
+                continue
+
+            prediction = predict(question, line)
+            if prediction == "TRUE":
+                st.success(line + "\n\nTRUE 😄")
+            else:
+                st.warning(line + "\n\nFALSE 😟")
+    
     # Create a prediction button
-    if st.button("Predict"):
+    elif st.button("Predict"):
+        prediction = ""
         stripped = text.strip()
         if not stripped:
-            st.error("Please enter some text.")
+            st.error("Please enter a contextext.")
             return
         stripped = question.strip()
         if not stripped:
             st.error("Please enter a question.")
             return
-        text = text.replace("\n", "")
+        
         prediction = predict(question, text)
         if prediction == "TRUE":
             st.success("TRUE 😄")