Update app.py

app.py

@@ -13,7 +13,7 @@ os.environ["OPENAI_API_KEY"] = os.environ['SECRET_CODE']
 
 # Need to write to persistent dataset because cannot store temp data on spaces
 DATASET_REPO_URL = "https://huggingface.co/datasets/peterpull/MediatorBot"
-DATA_FILENAME = "data.csv"
+DATA_FILENAME = "data.txt"
 DATA_FILE = os.path.join("data", DATA_FILENAME)
 
 # I am guessing we need a write access token.
@@ -28,36 +28,33 @@ repo = Repository(
 
 
 def generate_html() -> str:
-    with open(DATA_FILE) as csvfile:
-        reader = csv.DictReader(csvfile)
-        rows = []
-        for row in reader:
-            rows.append(row)
+    with open(DATA_FILE) as file:
+        rows = file.readlines()
         rows.reverse()
         if len(rows) == 0:
             return "no messages yet"
         else:
             html = "<div class='chatbot'>"
             for row in rows:
+                row_parts = row.split(",")
+                user = row_parts[0].strip()
+                chatbot = row_parts[1].strip()
+                time = row_parts[2].strip()
                 html += "<div>"
-                html += f"<span>{row['User']}</span>"
-                html += f"<span class='message'>{row['Chatbot']}</span>"
-                html += f"<span class='time'>{row['time']}</span>"
+                html += f"<span>{user}</span>"
+                html += f"<span class='message'>{chatbot}</span>"
+                html += f"<span class='time'>{time}</span>"
                 html += "</div>"
             html += "</div>"
             return html
 
 def store_message(chatinput: str, chatresponse: str):
     if chatinput and chatresponse:
-        with open(DATA_FILE, "a") as csvfile:
-            writer = csv.DictWriter(csvfile, fieldnames=["User", "Chatbot", "time"])
-            print({"User": chatinput, "Chatbot": chatresponse, "time": str(datetime.now())})
-            writer.writerow(
-                {"User": chatinput, "Chatbot": chatresponse, "time": str(datetime.now())}
-            )
+        with open(DATA_FILE, "a") as file:
+            file.write(f"User: {chatinput}, Chatbot: {chatresponse}, Time: {str(datetime.now())}\n")
+        # commit changes to repo
         commit_url = repo.push_to_hub()
         print(commit_url)
-
     return generate_html()