Update app.py

app.py

@@ -58,19 +58,21 @@ generator3 = gr.Interface.load("huggingface/EleutherAI/gpt-j-6B", api_key=HF_TOK
 
 SplitterInputBox = gr.inputs.Textbox(lines=5, label="Enter a sentence to get another sentence.")
 
-def AIMemory(name: str, message: str):
+def AIMemory(name: str):
     if name and message:
         with open(DATA_FILE, "a") as csvfile:
             writer = csv.DictWriter(csvfile, fieldnames=["name", "message", "time"])
             writer.writerow({"name": name, "message": message, "time": str(datetime.now())})
         commit_url = repo.push_to_hub()
-    return {"name": name, "message": message, "time": str(datetime.now())}
+        
+    return message
 
 
 with open('Mindfulness.txt', 'r') as file:
         context = file.read()
     
-parallelModel = gr.Parallel(generator1, generator2, generator3, 
+#parallelModel = gr.Parallel(generator1, generator2, generator3, 
+parallelModel = gr.Parallel(fn=AIMemory, generator1, generator2, generator3, 
 #inputs = SplitterInputBox, 
 inputs=[
             #gr.inputs.Textbox(lines=7, default=context, label=""),