ceckenrode commited on
Commit
830b90a
Β·
1 Parent(s): da53050

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -2
app.py CHANGED
@@ -143,6 +143,16 @@ def append_file(file_path, content):
143
  except:
144
  return "Error occurred while appending to file."
145
 
 
 
 
 
 
 
 
 
 
 
146
  title = """<h1 align="center">Memory Chat Story Generator ChatGPT</h1>"""
147
  description = """
148
  ## ChatGPT Datasets πŸ“š
@@ -192,13 +202,13 @@ with gr.Blocks(css = """#col_container {width: 1400px; margin-left: auto; margin
192
  saveFile = gr.Button("πŸ’Ύ Save File")
193
  deleteFile = gr.Button("πŸ—‘οΈ Delete File")
194
  appendFile = gr.Button("βž• Append File")
195
- appendFile = gr.Button("πŸ“₯ Download File")
196
  listFiles.click(list_files, inputs=fileName, outputs=fileContent)
197
  readFile.click(read_file, inputs=fileName, outputs=fileContent)
198
  saveFile.click(write_file, inputs=[fileName, fileContent], outputs=completedMessage)
199
  deleteFile.click(delete_file, inputs=fileName, outputs=completedMessage)
200
  appendFile.click(append_file, inputs=[fileName, fileContent], outputs=completedMessage )
201
- dowloadCSV.click(download_file, inputs=fileName, outputs=fileContent)
202
 
203
  inputs.submit(predict, [inputs, top_p, temperature,chat_counter, chatbot, state], [chatbot, state, chat_counter])
204
  b1.click(predict, [inputs, top_p, temperature, chat_counter, chatbot, state], [chatbot, state, chat_counter])
 
143
  except:
144
  return "Error occurred while appending to file."
145
 
146
+ def download_csv(file_path, headers, data):
147
+ try:
148
+ with open(file_path, "wb") as file:
149
+ writer = csv.writer(file)
150
+ writer.writerow(headers)
151
+ writer.writerows(data)
152
+ return f"Successfully downloaded {file_path}."
153
+ except:
154
+ return "Error occurred while downloading file."
155
+
156
  title = """<h1 align="center">Memory Chat Story Generator ChatGPT</h1>"""
157
  description = """
158
  ## ChatGPT Datasets πŸ“š
 
202
  saveFile = gr.Button("πŸ’Ύ Save File")
203
  deleteFile = gr.Button("πŸ—‘οΈ Delete File")
204
  appendFile = gr.Button("βž• Append File")
205
+ downloadFile = gr.Button("πŸ“₯ Download File")
206
  listFiles.click(list_files, inputs=fileName, outputs=fileContent)
207
  readFile.click(read_file, inputs=fileName, outputs=fileContent)
208
  saveFile.click(write_file, inputs=[fileName, fileContent], outputs=completedMessage)
209
  deleteFile.click(delete_file, inputs=fileName, outputs=completedMessage)
210
  appendFile.click(append_file, inputs=[fileName, fileContent], outputs=completedMessage )
211
+ download_csv.click(download_csv, inputs=fileName, outputs=fileContent)
212
 
213
  inputs.submit(predict, [inputs, top_p, temperature,chat_counter, chatbot, state], [chatbot, state, chat_counter])
214
  b1.click(predict, [inputs, top_p, temperature, chat_counter, chatbot, state], [chatbot, state, chat_counter])