Update app.py

app.py

@@ -40,8 +40,8 @@ def generate_html() -> str:
             html = "<div class='chatbot'>"
             for row in rows:
                 html += "<div>"
-                html += f"<span>{row['chatinput']}</span>"
-                html += f"<span class='message'>{row['chatresponse']}</span>"
+                html += f"<span>{row['User']}</span>"
+                html += f"<span class='message'>{row['Chatbot']}</span>"
                 html += f"<span class='time'>{row['time']}</span>"
                 html += "</div>"
             html += "</div>"
@@ -50,10 +50,10 @@ def generate_html() -> str:
 def store_message(chatinput: str, chatresponse: str):
     if chatinput and chatresponse:
         with open(DATA_FILE, "a") as csvfile:
-            writer = csv.writer(csvfile, delimiter=',', quotechar='"', quoting=csv.QUOTE_MINIMAL)
-            if csvfile.tell() == 0:
-                writer.writerow(['User input', 'Chatbot response', 'Date and Time'])
-            writer.writerow([chatinput, chatresponse, datetime.now().strftime('%Y-%m-%d %H:%M:%S')])
+            writer = csv.DictWriter(csvfile, fieldnames=["User", "Chatbot", "time"])
+            writer.writerow(
+                {"User": chatinput, "Chatbot": chatresponse, "time": str(datetime.now())}
+            )
         commit_url = repo.push_to_hub()
         print(commit_url)