Update app.py

app.py

@@ -1,9 +1,11 @@
 import os
 import csv
+from datetime import datetime
+
 import gradio as gr
 from gradio import components
 from huggingface_hub import Repository, hf_hub_download
-from datetime import datetime
+
 
 DATASET_REPO_URL = "https://huggingface.co/datasets/sjw/data.csv"
 DATASET_REPO_ID = "sjw/data.csv"
@@ -20,44 +22,44 @@ try:
         force_filename=DATA_FILENAME
     )
 except:
-    print("file not found")
+    print("File not found")
+
 
 repo = Repository(
-    local_dir = DIRNAME, clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
+    local_dir=DIRNAME, 
+    clone_from=DATASET_REPO_URL, 
+    use_auth_token=HF_TOKEN
 )
 
 
 def generate_html() -> str:
+    """Generate HTML content for the chat."""
     with open(DATA_FILE) as csvfile:
         reader = csv.DictReader(csvfile)
-        rows = []
-        for row in reader:
-            rows.append(row)
+        rows = [row for row in reader]
         rows.reverse()
-        if len(rows) == 0:
-            return "no messages yet"
-        else:
-            html = "<div class='chatbot'>"
-            for row in rows:
-                html += "<div>"
-                html += f"<span><b>{row['name']}</b></span> "  # Make the name bold and add a space after it
-                html += f"<span class='message'>{row['message']}</span>"
-                html += "</div>"
-            html += "</div>"
-            return html
-        
-def store_message(name: str, message: str):
 
+        if not rows:
+            return "No messages yet"
+
+        html = "<div class='chatbot'>"
+        for row in rows:
+            html += f"<div><span><b>{row['name']}</b></span> <span class='message'>{row['message']}</span></div>"
+        html += "</div>"
+        return html
+
+
+def store_message(name: str, message: str) -> str:
+    """Store the message and regenerate HTML content."""
     if name and message:
         with open(DATA_FILE, "a") as csvfile:
             writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
-            writer.writerow(
-                {"name": name, "message": message, "time": str(datetime.now())}
-            )
+            writer.writerow({"name": name, "message": message, "time": str(datetime.now())})
     return generate_html()
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Title")
+
+with gr.Blocks() as feedback_page:
+    gr.Markdown("# User Feedback")
     name_input = components.Textbox(label="Your Username")
     message_input = components.Textbox(label="Your Feedback", lines=2)
     output_html = gr.HTML()
@@ -65,4 +67,4 @@ with gr.Blocks() as demo:
 
     submit_button.click(store_message, inputs=[name_input, message_input], outputs=output_html)
 
-demo.launch()
+feedback_page.launch()