Update app.py

app.py

@@ -86,6 +86,64 @@ def predict(inputs, top_p, temperature, chat_counter, chatbot=[], history=[]):
 def reset_textbox():
     return gr.update(value='')
 
+
+    
+
+# Episodic and Semantic IO
+def list_files(file_path):
+    import os
+    icon_csv = "📄 "
+    icon_txt = "📑 "
+    current_directory = os.getcwd()
+    file_list = []
+    for filename in os.listdir(current_directory):
+        if filename.endswith(".csv"):
+            file_list.append(icon_csv + filename)
+        elif filename.endswith(".txt"):
+            file_list.append(icon_txt + filename)
+    if file_list:
+        return "\n".join(file_list)
+    else:
+        return "No .csv or .txt files found in the current directory."
+
+# Function to read a file
+def read_file(file_path):
+    try:
+        with open(file_path, "r") as file:
+            contents = file.read()
+            return f"{contents}"
+            #return f"Contents of {file_path}:\n{contents}"
+    except FileNotFoundError:
+        return "File not found."
+
+# Function to delete a file
+def delete_file(file_path):
+    try:
+        import os
+        os.remove(file_path)
+        return f"{file_path} has been deleted."
+    except FileNotFoundError:
+        return "File not found."
+
+# Function to write to a file
+def write_file(file_path, content):
+    try:
+        with open(file_path, "w") as file:
+            file.write(content)
+        return f"Successfully written to {file_path}."
+    except:
+        return "Error occurred while writing to file."
+
+# Function to append to a file
+def append_file(file_path, content):
+    try:
+        with open(file_path, "a") as file:
+            file.write(content)
+        return f"Successfully appended to {file_path}."
+    except:
+        return "Error occurred while appending to file."
+
+
 title = """<h1 align="center">Memory Chat Story Generator ChatGPT</h1>"""
 description = """
 ## ChatGPT Datasets 📚
@@ -123,6 +181,25 @@ with gr.Blocks(css = """#col_container {width: 1400px; margin-left: auto; margin
             temperature = gr.Slider( minimum=-0, maximum=5.0, value=1.0, step=0.1, interactive=True, label="Temperature",)
             chat_counter = gr.Number(value=0, visible=True, precision=0)
 
+            
+    # Episodic/Semantic IO
+    fileName = gr.Textbox(label="Filename")
+    fileContent = gr.TextArea(label="File Content")
+    completedMessage = gr.Textbox(label="Completed")
+    label = gr.Label()
+    with gr.Row():
+        listFiles = gr.Button("📄 List File(s)")
+        readFile = gr.Button("📖 Read File")
+        saveFile = gr.Button("💾 Save File")
+        deleteFile = gr.Button("🗑️ Delete File")
+        appendFile = gr.Button("➕ Append File")
+    listFiles.click(list_files, inputs=fileName, outputs=fileContent)
+    readFile.click(read_file, inputs=fileName, outputs=fileContent)
+    saveFile.click(write_file, inputs=[fileName, fileContent], outputs=completedMessage)
+    deleteFile.click(delete_file, inputs=fileName, outputs=completedMessage)
+    appendFile.click(append_file, inputs=[fileName, fileContent], outputs=completedMessage )
+
+    
     inputs.submit(predict, [inputs, top_p, temperature,chat_counter, chatbot, state], [chatbot, state, chat_counter])
     b1.click(predict, [inputs, top_p, temperature, chat_counter, chatbot, state], [chatbot, state, chat_counter])
     b1.click(reset_textbox, [], [inputs])