Update app.py

app.py

@@ -20,8 +20,7 @@ from tenacity import (
     wait_random_exponential,
 )
 import bf_trigger
-# import change_ui as change
-import offline_update
+import chat_engine as chat_gen
 # for exponential backoff
 
 # FUNCTIONS
@@ -893,6 +892,13 @@ def upload_file(files):
     file_paths = [file.name for file in files]
     return file_paths
 
+chat_engine = chat_gen.get_chat_engine(files)
+
+def chatbot_response(message, history):
+    stream = chat_engine.stream_chat(message, chat_history=history)
+    for text in stream.response_gen:
+        yield text
+
 collection_names = get_collection_names()
 
 with gr.Blocks() as block:
@@ -903,8 +909,7 @@ with gr.Blocks() as block:
     
     with gr.Column(visible=True, elem_id='login') as login:
         password = gr.Textbox(label='Enter Password')
-        # with gr.Row():
-        #     login_to_zipfile_btn = gr.Button("Login")
+
     with gr.Tab("Zipfile Upload"):
         with gr.Column(visible=True, elem_id='zipfile') as zipfile:
             with gr.Row():
@@ -922,10 +927,7 @@ with gr.Blocks() as block:
                     upload = gr.Label(label='Uploader')
                     err_imgs = gr.Textbox(label="Images Not Processed")
                     stop = gr.Button("Stop Run")
-                    # with gr.Row():
-                    #     zipfile_to_login_btn = gr.Button("Back to Login")
-                    #     zipfile_to_trigger_btn = gr.Button("Brandfolder Trigger")
-                    #     zipfile_to_offline_btn = gr.Button("Offline Image Update")
+
     with gr.Tab("Brandfolder AI Trigger"):
         with gr.Column(visible=True, elem_id='trigger') as trigger:
             gr.Markdown('''
@@ -941,9 +943,17 @@ with gr.Blocks() as block:
             bf_button = gr.Button('Run AI algorithm for Pre-Processed Images')
             bf_upload = gr.Label(label='Uploader')
             stop_bf = gr.Button('Stop Run')
-            # with gr.Row():
-            #     trigger_to_zipfile_btn = gr.Button("Zipfile Upload")
-            #     trigger_to_offline_btn = gr.Button("Offline Image Update")
+
+
+    with gr.Tab("DNA Chatbot"):
+        with gr.Column():
+            gr.Markdown('''
+            # DNA Chatbot
+            This DNA chatbot uses the uploaded dna documents to answer questions
+            ''')
+            chatbot = gr.Chatbot()
+            msg = gr.Textbox()
+            clear = gr.ClearButton([msg, chatbot])
 
     # with gr.Column(visible=False, elem_id='offline') as offline:
     #     gr.Markdown('''
@@ -958,11 +968,10 @@ with gr.Blocks() as block:
     #     offline_button = gr.Button('Run AI algorithm for AI Processed Images Section')
     #     offline_upload = gr.Label(label='Uploader')
     #     stop_offline = gr.Button("Stop Run")
-    #     # with gr.Row():
-    #     #     offline_to_zipfile_btn = gr.Button("Zipfile Upload")
-    #     #     offline_to_trigger_btn = gr.Button("Brandfolder Trigger")
+
 
     # selection.select(fn=get_collection_names, outputs=[selection])
+    msg.sumbit(chatbot_response, [msg, chatbot], [msg, chatbot])
     algo_event = algorithm.click(fn=import_client_data, inputs=[selection, zipfile, topical_map, password, project_bool, ai_bool], outputs=[upload, err_imgs])
     bf_event = bf_button.click(fn=bf_trigger.run_preprocess_ai, inputs=[bf_topical_map, bf_selection, section], outputs=[bf_upload])
     # offline_event = offline_button.click(fn=offline_update.run_preprocess_ai, inputs=[offline_topical_map, offline_selection], outputs=[offline_upload])
@@ -970,61 +979,6 @@ with gr.Blocks() as block:
     stop_bf.click(fn=None, inputs=None, outputs=None, cancels=[bf_event])
     upload_btn.upload(upload_file, upload_btn, zipfile)
     # stop_offline.click(fn=None, inputs=None, outputs=None, cancels=[offline_event])
-    # login_to_zipfile_btn.click(
-    #     fn=None,
-    #     inputs=None,
-    #     outputs=[login, zipfile],
-    #     js=change.login_to_zipfile_js
-    # )
-    
-    # zipfile_to_login_btn.click(
-    #     fn=None,
-    #     inputs=None,
-    #     outputs=[login, zipfile],
-    #     js=change.zipfile_to_login_js
-    # )
-    
-    
-    # zipfile_to_trigger_btn.click(
-    #     fn=None,
-    #     inputs=None,
-    #     outputs=[zipfile, trigger],
-    #     js=change.zipfile_to_trigger_js
-    # )
-    
-    # trigger_to_zipfile_btn.click(
-    #     fn=None,
-    #     inputs=None,
-    #     outputs=[zipfile, trigger],
-    #     js=change.trigger_to_zipfile_js
-    # )
-    
-    # zipfile_to_offline_btn.click(
-    #     fn=None,
-    #     inputs=None,
-    #     outputs=[zipfile, offline],
-    #     js=change.zipfile_to_offline_js
-    # )
-    
-    # trigger_to_offline_btn.click(
-    #     fn=None,
-    #     inputs=None,
-    #     outputs=[trigger, offline],
-    #     js=change.trigger_to_offline_js
-    # )
-    
-    # offline_to_zipfile_btn.click(
-    #     fn=None,
-    #     inputs=None,
-    #     outputs=[zipfile, offline],
-    #     js=change.offline_to_zipfile_js
-    # )
-    
-    # offline_to_trigger_btn.click(
-    #     fn=None,
-    #     inputs=None,
-    #     outputs=[trigger, offline],
-    #     js=change.offline_to_trigger_js
-    # )
+   
 block.queue(default_concurrency_limit=5)
 block.launch()