Update app.py

app.py

@@ -113,35 +113,36 @@ def features(sentence, index):
         'is_numeric': sentence[index].isdigit(),
     }
 
-# Function to prepare text
-def prepare_text(text):
-    symbol_pattern = r'([^\w\s\d])'
-    prepared_text = re.sub(symbol_pattern, r' \1 ', text)
-    prepared_text = re.sub(r'\s+', ' ', prepared_text)
-    return prepared_text.strip()
-
-# Text input field for user input
-text_input = st.text_input("Enter some text:")
-
-# Check if the user input is not empty
-if text_input:
-    # Prepare text
-    prepared_text = prepare_text(text_input)
-    
-    # Tokenize text
-    tokenized_text = word_tokenize(prepared_text)
-    
-    # Extract features
-    features_list = [features(tokenized_text, i) for i in range(len(tokenized_text))]
-    
-    # Create a DataFrame with the features
-    data = pd.DataFrame(features_list)
-    
-    # Load the model from the Hub
-    model_id = "Alshargi/arabic-msa-dialects-segmentation"
-    res = hub_utils.get_model_output(model_id, data)
-    
-    # Display the model output
-    st.write("Model Output:", res)
-else:
-    st.write("Please enter some text.")
+import gradio as gr
+
+# Define the function for processing user input
+def process_text(text_input):
+    if text_input:
+        # Prepare text
+        prepared_text = prepare_text(text_input)
+        
+        # Tokenize text
+        tokenized_text = word_tokenize(prepared_text)
+        
+        # Extract features
+        features_list = [features(tokenized_text, i) for i in range(len(tokenized_text))]
+        
+        # Create a DataFrame with the features
+        data = pd.DataFrame(features_list)
+        
+        # Load the model from the Hub
+        model_id = "Alshargi/arabic-msa-dialects-segmentation"
+        res = hub_utils.get_model_output(model_id, data)
+        
+        # Return the model output
+        return res
+    else:
+        return "Please enter some text."
+
+# Define the Gradio interface
+iface = gr.Interface(fn=process_text, inputs="text", outputs="text", title="Arabic Text Segmentation")
+
+# Launch the Gradio interface
+iface.launch()
+
+