Update app.py

app.py

@@ -27,6 +27,7 @@ def SaveResult(text, outputfileName):
             f.write('\n')
     else:
         with open(outputfileName, "w") as f: #write
+            f.write(str("time, message, text\n")) # one time only to get column headers for CSV file
             f.write(str(text.replace("\n","  ")))
             f.write('\n')
     return
@@ -66,12 +67,18 @@ title = "💬ChatBack🧠💾"
 description = """Chatbot With persistent memory dataset allowing multiagent system AI to access a shared dataset as memory pool with stored interactions. 
  Current Best SOTA Chatbot:  https://huggingface.co/facebook/blenderbot-400M-distill?text=Hey+my+name+is+ChatBack%21+Are+you+ready+to+rock%3F  """
 
+def get_base(filename): 
+        basedir = os.path.dirname(__file__)
+        loadPath = basedir + "\\" + filename
+        return loadPath
+    
 def chat(message, history):
     history = history or []
     if history: 
         history_useful = ['</s> <s>'.join([str(a[0])+'</s> <s>'+str(a[1]) for a in history])]
     else:
         history_useful = []
+        
     history_useful = add_note_to_history(message, history_useful)
     inputs = tokenizer(history_useful, return_tensors="pt")
     inputs, history_useful, history = take_last_tokens(inputs, history_useful, history)
@@ -86,15 +93,13 @@ def chat(message, history):
     if UseMemory: 
         outputfileName = 'File.csv'
         df = store_message(message, response, outputfileName) # Save to dataset
-        basedir = os.path.dirname(__file__)
-        savePath = outputfileName
+        basedir = get_base(outputfileName)
         
-    #return history, df, outputfileName
-    return history, df
+    return history, df, basedir
 
+    
 with gr.Blocks() as demo:
-  gr.Markdown("<h1><center>🍰Gradio chatbot backed by memory in a dataset repository.🎨</center></h1>")
-  #gr.Markdown("The memory dataset for saves is [{DATASET_REPO_URL}]({DATASET_REPO_URL}) And here: https://huggingface.co/spaces/awacke1/DatasetAnalyzer  Code and datasets on chat are here  hf  tk: https://paperswithcode.com/datasets?q=chat&v=lst&o=newest")
+  gr.Markdown("<h1><center>🍰Gradio chatbot backed by memory in a local CSV file.🎨</center></h1>")
   
   with gr.Row():
     t1 = gr.Textbox(lines=1, default="", label="Chat Text:")
@@ -102,13 +107,11 @@ with gr.Blocks() as demo:
     
   with gr.Row(): # inputs and buttons
     s1 = gr.State([])
-    s2 = gr.Markdown()
-  with gr.Row():
-    file = gr.File(label="File"),
     df1 = gr.Dataframe(wrap=True, max_rows=1000, overflow_row_behaviour= "paginate")
+  with gr.Row(): # inputs and buttons
+    file = gr.File(label="File")
+    s2 = gr.Markdown()
 
-      
-  #b1.click(fn=chat, inputs=[t1, s1], outputs=[s1, df1, file]) 
-  b1.click(fn=chat, inputs=[t1, s1], outputs=[s1, df1]) 
+  b1.click(fn=chat, inputs=[t1, s1], outputs=[s1, df1, file]) 
 
 demo.launch(debug=True, show_error=True)