Update app.py

app.py

@@ -11,7 +11,6 @@ import csv
 
 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.txt"
@@ -20,19 +19,31 @@ DATA_FILE = os.path.join("data", DATA_FILENAME)
 # I am guessing we need a write access token.
 HF_TOKEN = os.environ.get("HF_TOKEN")
 print("HF TOKEN is none?", HF_TOKEN is None)
+print("HF hub ver", huggingface_hub.__version__)
 
 repo = Repository(
     local_dir="data", 
     clone_from=DATASET_REPO_URL, 
     use_auth_token=HF_TOKEN)
 
+
+def generate_text() -> str:
+    with open(DATA_FILE) as file:
+        text = ""
+        for line in file:
+            row_parts = line.strip().split(";")
+            if len(row_parts) != 3:
+                continue
+            user, chatbot, time = row_parts
+            text += f"Time: {time}\nUser: {user}\nChatbot: {chatbot}\n\n"
+        return text if text else "No messages yet"
+
 def store_message(chatinput: str, chatresponse: str):
     if chatinput and chatresponse:
         with open(DATA_FILE, "a") as file:
-            now = datetime.now()
-            timestamp = now.strftime("%Y-%m-%d %H:%M:%S")
-            file.write(f"{timestamp},{chatinput},{chatresponse}\n")
-            print(f"Stored message: {timestamp},{chatinput},{chatresponse}")
+            file.write(f"{datetime.now()},{chatinput},{chatresponse}\n")
+
+    return generate_text()
 
             
 #gets the index file which is the context data