freqlist / app.py
MK-316's picture
Create app.py
da92bd2 verified
raw
history blame
1.53 kB
import gradio as gr
from collections import Counter
import string
def process_text(text, sorting_option):
# Remove punctuation from the input text
translator = str.maketrans('', '', string.punctuation)
text = text.translate(translator)
words = text.split()
# Count word frequencies
word_counts = Counter(words)
# Sort words based on the selected option
if sorting_option == 'alphabetically':
sorted_words = sorted(word_counts.items(), key=lambda x: x[0])
elif sorting_option == 'by_frequency':
sorted_words = sorted(word_counts.items(), key=lambda x: x[1], reverse=True)
else:
sorted_words = word_counts.items()
# Get the top 5 words with their frequencies
top_5_words = sorted_words
# Format the top 5 words and frequencies as an HTML table
table_html = "<table style='width:100%'><tr><th>Word</th><th>Frequency</th></tr>"
for word, freq in top_5_words:
table_html += f"<tr><td>{word}</td><td>{freq}</td></tr>"
table_html += "</table>"
# Wrap the table in a div with a fixed height and scrolling
div_with_scroll = f"<div style='height: 200px; overflow-y: scroll;'>{table_html}</div>"
return div_with_scroll
iface = gr.Interface(
fn=process_text,
inputs=[gr.Textbox("text", label="Paste Text Here"),
gr.Radio(["alphabetically", "by_frequency", "none"], label="Select Sorting Option")],
outputs=gr.HTML(label="Top 5 Words with Frequencies")
)
iface.launch(share=True)