Update app.py

app.py

@@ -113,6 +113,39 @@ def features(sentence, index):
     }
 
 
+def prepare_text(text):
+    # Define regular expression pattern to match symbols and punctuation from any language
+    symbol_pattern = r'([^\w\s\d])'  # Capture non-word, non-space, non-digit characters
+    prepared_text = re.sub(symbol_pattern, r' \1 ', text)
+    prepared_text = re.sub(r'\s+', ' ', prepared_text)
+
+    return prepared_text.strip()  # Remove leading and trailing spaces
+
+
+
+def rebuildxx(ww, xres):
+    numprfx = xres.count('p')
+    numsufx = xres.count('f')
+    resfinal = ''
+    if numprfx != 0 and  numsufx != 0 :
+        resfinal = "{}+{}+{}".format(ww[:numprfx] , ww[numprfx:-numsufx] ,  ww[-numsufx:] )
+    if numprfx == 0 and  numsufx == 0 :
+        #resfinal = "{}+{}+{}".format("", ww , ""  )
+        resfinal = "{}".format(ww )
+
+    if numprfx == 0 and  numsufx != 0 :
+        #resfinal = "{}+{}+{}".format("" , ww[:-numsufx], ww[-numsufx:] )
+        resfinal = "{}+{}".format(ww[:-numsufx], ww[-numsufx:] )
+
+    if numprfx != 0 and  numsufx == 0 :
+        #resfinal = "{}+{}+{}".format(ww[:numprfx] ,  ww[numprfx:],  "")
+        resfinal = "{}+{}".format(ww[:numprfx] ,  ww[numprfx:])
+
+    return resfinal
+
+
+    
+
 # Define the function for processing user input
 def process_text(text_input):
     if text_input:
@@ -141,7 +174,7 @@ def process_text(text_input):
         return "Please enter some text."
 
 def main():
-    st.title("Model Output with Streamlit")
+    st.title("Arabic segmintation Model Output with Streamlit")
     
     # Text input
     input_text = st.text_input("Enter your text:")
@@ -149,8 +182,14 @@ def main():
     # Process the text when a button is clicked
     if st.button("Process"):
         output = process_text(input_text)
+        gg = word_tokenize(prepare_text(input_text)))
+        cc = ""
+        for x, y in zip(gg, output):
+            cc += rebuildxx(x, y) + " "
+
+        #print(cc)
         st.write("Model Output:")
-        st.write(output)
+        st.write(cc)
 
 if __name__ == "__main__":
     main()