mylesai commited on
Commit
b858839
·
verified ·
1 Parent(s): 74c86bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -902,10 +902,11 @@ def generate_content(csv_file, query_engine):
902
  print(csv_file)
903
  df = con_gen.get_content_csv(csv_file, query_engine[-1])
904
  print(df.head(10)['Content'])
905
- output_file = df.to_csv('./output.csv')
 
906
  print(output_file)
907
  completion_status = "Done"
908
- return completion_status, gr.DownloadButton(label=f"Download Content", value=output_file, visible=True)
909
 
910
  collection_names = get_collection_names()
911
 
@@ -994,8 +995,9 @@ with gr.Blocks() as block:
994
  ''')
995
  website_layout_file = gr.File(label='Website Layout File')
996
  con_gen_btn = gr.Button('Generate Content')
 
997
  status = gr.Textbox(label='Completion Status')
998
- download_btn = gr.DownloadButton(label='Download File', visible=False)
999
 
1000
 
1001
 
@@ -1020,7 +1022,7 @@ with gr.Blocks() as block:
1020
  msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
1021
  bot, [chatbot, chat_engine], chatbot)
1022
  clear.click(lambda: None, None, chatbot, queue=False)
1023
- con_gen_btn.click(generate_content, inputs=[website_layout_file, query_engine], outputs=[status, download_btn])
1024
  algo_event = algorithm.click(fn=import_client_data, inputs=[selection, zipfile, topical_map, password, project_bool, ai_bool], outputs=[upload, err_imgs])
1025
  bf_event = bf_button.click(fn=bf_trigger.run_preprocess_ai, inputs=[bf_topical_map, bf_selection, section], outputs=[bf_upload])
1026
  # offline_event = offline_button.click(fn=offline_update.run_preprocess_ai, inputs=[offline_topical_map, offline_selection], outputs=[offline_upload])
 
902
  print(csv_file)
903
  df = con_gen.get_content_csv(csv_file, query_engine[-1])
904
  print(df.head(10)['Content'])
905
+ data_preview = df.head(10)
906
+ output_file = df.to_csv('output.csv')
907
  print(output_file)
908
  completion_status = "Done"
909
+ return completion_status, data_preview, gr.File(label="Download Content", value=output_file, visible=True)
910
 
911
  collection_names = get_collection_names()
912
 
 
995
  ''')
996
  website_layout_file = gr.File(label='Website Layout File')
997
  con_gen_btn = gr.Button('Generate Content')
998
+ data_preview = gr.DataFrame(label='Processed DataFrame Preview')
999
  status = gr.Textbox(label='Completion Status')
1000
+ download_btn = gr.File(label='Download Content', visible=False)
1001
 
1002
 
1003
 
 
1022
  msg.submit(user, [msg, chatbot], [msg, chatbot], queue=False).then(
1023
  bot, [chatbot, chat_engine], chatbot)
1024
  clear.click(lambda: None, None, chatbot, queue=False)
1025
+ con_gen_btn.click(generate_content, inputs=[website_layout_file, query_engine], outputs=[status, data_preview, download_btn])
1026
  algo_event = algorithm.click(fn=import_client_data, inputs=[selection, zipfile, topical_map, password, project_bool, ai_bool], outputs=[upload, err_imgs])
1027
  bf_event = bf_button.click(fn=bf_trigger.run_preprocess_ai, inputs=[bf_topical_map, bf_selection, section], outputs=[bf_upload])
1028
  # offline_event = offline_button.click(fn=offline_update.run_preprocess_ai, inputs=[offline_topical_map, offline_selection], outputs=[offline_upload])