Update app.py

app.py

@@ -35,29 +35,62 @@ def process_text(text, operation, correct_spacing, remove_diacritics, remove_spe
         pos_tags = pos_tagger.tag(tokens)  # Generate POS tags
         chunks = chunker.parse(pos_tags)  # Pass tagged tokens to Chunker
         result = str(chunks)  # Show chunks as text
-    
+    elif operation == "pos_tag":
+        tokens = word_tokenize(text)
+        pos_tags = pos_tagger.tag(tokens)
+        result = " ".join([f"{token}/{tag}" for token, tag in pos_tags])  # Format: token/POS
+
     return result
 
+def toggle_normalization_options(operation):
+    # Show normalization options only if 'normalize' is selected
+    is_normalize = (operation == "normalize")
+    return [gr.update(visible=is_normalize)] * 9  # Update visibility for all 9 checkboxes
+
 # Define Gradio interface
-operations = ['normalize', 'tokenize', 'lemmatize', 'chunk']
-iface = gr.Interface(
-    fn=process_text,
-    inputs=[
-        gr.Textbox(lines=10, label="Input Text"),
-        gr.Radio(operations, label="Select Operation", type="value"),  # Radio button to select one operation at a time
-        gr.Checkbox(value=True, label="Correct Spacing", interactive=True),
-        gr.Checkbox(value=True, label="Remove Diacritics", interactive=True),
-        gr.Checkbox(value=True, label="Remove Special Characters", interactive=True),
-        gr.Checkbox(value=True, label="Decrease Repeated Characters", interactive=True),
-        gr.Checkbox(value=True, label="Persian Style", interactive=True),
-        gr.Checkbox(value=True, label="Persian Numbers", interactive=True),
-        gr.Checkbox(value=True, label="Unicodes Replacement", interactive=True),
-        gr.Checkbox(value=True, label="Separate 'می'", interactive=True)
-    ],
-    outputs=gr.Textbox(label="Processed Text", interactive=False, lines=10, show_copy_button=True, show_label=True),  # Output as copyable text
-    title="Persian Text Processor with Hazm",
-    description="Select an operation and normalization parameters to process the input text using Hazm."
-)
+with gr.Blocks() as demo:
+    gr.Markdown("# Persian Text Processor with Hazm")
+    gr.Markdown("Select an operation and, if applicable, adjust normalization parameters to process the input text using Hazm.")
+    
+    with gr.Row():
+        input_text = gr.Textbox(lines=10, label="Input Text")
+    
+    with gr.Row():
+        operation = gr.Radio(
+            choices=['normalize', 'tokenize', 'lemmatize', 'chunk', 'pos_tag'],
+            label="Select Operation",
+            value='normalize'
+        )
+    
+    with gr.Column(visible=True) as normalization_options:
+        correct_spacing = gr.Checkbox(value=True, label="Correct Spacing")
+        remove_diacritics = gr.Checkbox(value=True, label="Remove Diacritics")
+        remove_specials_chars = gr.Checkbox(value=True, label="Remove Special Characters")
+        decrease_repeated_chars = gr.Checkbox(value=True, label="Decrease Repeated Characters")
+        persian_style = gr.Checkbox(value=True, label="Persian Style")
+        persian_numbers = gr.Checkbox(value=True, label="Persian Numbers")
+        unicodes_replacement = gr.Checkbox(value=True, label="Unicodes Replacement")
+        seperate_mi = gr.Checkbox(value=True, label="Separate 'می'")
+    
+    operation.change(
+        fn=toggle_normalization_options,
+        inputs=operation,
+        outputs=normalization_options
+    )
+    
+    output_text = gr.Textbox(label="Processed Text", lines=10, interactive=False, show_copy_button=True)
+    
+    submit_button = gr.Button("Process Text")
+    submit_button.click(
+        fn=process_text,
+        inputs=[
+            input_text, operation,
+            correct_spacing, remove_diacritics, remove_specials_chars,
+            decrease_repeated_chars, persian_style, persian_numbers,
+            unicodes_replacement, seperate_mi
+        ],
+        outputs=output_text
+    )
 
 if __name__ == "__main__":
-    iface.launch()
+    demo.launch()