Update app.py

app.py

@@ -10,7 +10,8 @@ import fastapi
 from typing import List, Dict
 import httpx
 import pandas as pd
-import datasets as ds
+
+
 UseMemory=True
 
 HF_TOKEN=os.environ.get("HF_TOKEN")
@@ -36,12 +37,28 @@ def SaveResult(text, outputfileName):
 def store_message(name: str, message: str, outputfileName: str):
     basedir = os.path.dirname(__file__)
     savePath = outputfileName
-    if name and message:
-        with open(savePath, "a") as csvfile:
-            writer = csv.DictWriter(csvfile, fieldnames=[ "time", "message", "name", ])
-            writer.writerow(
-                {"time": str(datetime.now()), "message": message.strip(), "name": name.strip()  }
-            )
+    
+    # if file doesnt exist, create it with labels
+    from os.path import exists
+    file_exists = exists(savePath)
+    
+    if (file_exists==False):
+        with open(savePath, "w") as f: #write
+            f.write(str("time, message, text\n")) # one time only to get column headers for CSV file
+            if name and message:
+                writer = csv.DictWriter(f, fieldnames=["time", "message", "name"])
+                writer.writerow(
+                    {"time": str(datetime.now()), "message": message.strip(), "name": name.strip()  }
+                )
+        df = pd.read_csv(savePath)
+        df = df.sort_values(df.columns[0],ascending=False)
+    else:
+        if name and message:
+            with open(savePath, "a") as csvfile:
+                writer = csv.DictWriter(csvfile, fieldnames=[ "time", "message", "name", ])
+                writer.writerow(
+                    {"time": str(datetime.now()), "message": message.strip(), "name": name.strip()  }
+                )
         df = pd.read_csv(savePath)
         df = df.sort_values(df.columns[0],ascending=False)
     return df
@@ -69,7 +86,10 @@ description = """Chatbot With persistent memory dataset allowing multiagent syst
 
 def get_base(filename): 
         basedir = os.path.dirname(__file__)
-        loadPath = basedir + "\\" + filename
+        print(basedir)
+        #loadPath = basedir + "\\" + filename # works on windows
+        loadPath = basedir + filename 
+        print(loadPath)
         return loadPath
     
 def chat(message, history):
@@ -91,7 +111,7 @@ def chat(message, history):
     df=pd.DataFrame()
     
     if UseMemory: 
-        outputfileName = 'File.csv'
+        outputfileName = 'ChatbotMemory01-24-2023-05-07-PM.csv'
         df = store_message(message, response, outputfileName) # Save to dataset
         basedir = get_base(outputfileName)
         
@@ -99,11 +119,11 @@ def chat(message, history):
 
     
 with gr.Blocks() as demo:
-  gr.Markdown("<h1><center>🍰Gradio chatbot backed by memory in a local CSV file.🎨</center></h1>")
+  gr.Markdown("<h1><center>🍰Gradio chatbot backed by dataframe CSV memory🎨</center></h1>")
   
   with gr.Row():
     t1 = gr.Textbox(lines=1, default="", label="Chat Text:")
-    b1 = gr.Button("Send Message")
+    b1 = gr.Button("Respond and Retrieve Messages")
     
   with gr.Row(): # inputs and buttons
     s1 = gr.State([])