Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from collections import Counter
 import string
+import csv
+import io
 
 def process_text(text, sorting_option):
     # Remove punctuation from the input text
@@ -29,16 +31,31 @@ def process_text(text, sorting_option):
         table_html += f"<tr><td>{word}</td><td>{freq}</td></tr>"
     table_html += "</table>"
     
+    # Create a CSV file
+    csv_data = []
+    for word, freq in top_5_words:
+        csv_data.append([word, freq])
+    
+    # Write CSV data to a string buffer
+    csv_buffer = io.StringIO()
+    csv_writer = csv.writer(csv_buffer)
+    csv_writer.writerow(["Word", "Frequency"])
+    csv_writer.writerows(csv_data)
+    csv_buffer.seek(0)
+    
+    # Create a download link for the CSV file
+    csv_download_link = f"<a href='data:application/csv;charset=utf-8,{csv_buffer.getvalue()}' download='word_frequencies.csv'>Download CSV</a>"
+    
     # 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
+    return div_with_scroll, csv_download_link
 
 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")
+    outputs=[gr.HTML(label="Top 5 Words with Frequencies"), gr.HTML(label="Download CSV")]
 )
 
 iface.launch(share=True)