Update app.py

app.py

@@ -16,7 +16,6 @@ def getBrains():
     return gr.update(choices=active_indexes)
 
 
-
 prompt_names = ["Industry / Topic Research", "Data Analysis",
                 "Rewrite / Repurpose Content", "Create Long Form Article/Blog"]
 prompt_enhancers = ["""Write at is if you had a Harvard MBA. You are providing in-depth report that is captured at the end of this [PROMPT]
@@ -68,29 +67,32 @@ mycss = """
     
     #question {{background-color: {bgcolor};border-style:none; !important;box-shadow:none !important;padding-left:0;padding-right:0}}
     #question span {{background-color:{bgcolor};color:{scolor};font-size:18px}}
-    #output {{background-color: {bgcolor};border-style:none;border-width:0;box-shadow:none}}
+    #output {{background-color: {bgcolor};border-style:none;!important;box-shadow:none !important;padding-left:0;padding-right:0}}
     #output span {{background-color:{bgcolor};color:{scolor};font-size:18px}}
     #temp span {{background-color:#ffffff;color:{scolor}}}
     #temp input {{accent-color:{scolor}}}
     #tokens span {{background-color:#ffffff;color:{scolor}}}
     #tokens input {{accent-color:{scolor}}}
-    #button {{background-color:{scolor};color:#ffffff;margin-top:29px}}
+    #button {{background-color:{scolor};color:#ffffff;margin-top:22px}}
+    #buttonfollowup {{background-color:{scolor};color:#ffffff;margin-top:10px}}
 """
 formatted_css = mycss.format(bgcolor=bg_color, scolor=s_color)
 
 
 def handleSubmit(brain_name, enhancer, question, temperature, maxTokens):
     print(brain_name)
+    prompt = ""
     if (brain_name == "" and question == ""):
-        return "Please select Brain Name & Enter Question"
+        return "Please select Brain Name & Enter Question", False
     if (brain_name == ""):
-        return "Please select Brain Name"
+        return "Please select Brain Name", False
     if (question == ""):
-        return "Please Enter Question"
+        return "Please Enter Question", False
     if (enhancer != ""):
         promptIndex = prompt_names.index(enhancer)
-        question = prompt_enhancers[promptIndex]+question
-    return askQuestion(brain_name, question, temperature, maxTokens)
+        prompt = prompt_enhancers[promptIndex]
+        # question = prompt_enhancers[promptIndex]+question
+    return askQuestion(brain_name, question, prompt, temperature, maxTokens)
 
 
 with gr.Blocks(theme=gr.themes.Soft(), css=formatted_css) as block_demo:
@@ -104,6 +106,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=formatted_css) as block_demo:
 
     with gr.Row(elem_id="secondrow"):
 
+        memory = gr.State(value=False)
+
         with gr.Column(scale=1,  elem_id="inputsCol"):
 
             brain_name = gr.Dropdown(
@@ -119,15 +123,20 @@ with gr.Blocks(theme=gr.themes.Soft(), css=formatted_css) as block_demo:
                 maxTokens = gr.Slider(minimum=200, maximum=2000,
                                       step=100, value=1000, label="Max Tokens", elem_id="tokens")
 
-            submit_button = gr.Button(label="Submit", elem_id="button")
+            submit_button = gr.Button(value="Submit", elem_id="button")
 
-        with gr.Column(scale=1, elem_id="outputCol",):
+        with gr.Column(scale=1, elem_id="outputCol"):
             output_text = gr.TextArea(
-                label="Brain Output", lines=17, elem_id="output").style(show_copy_button=True)
+                label="Brain Output", lines=13, elem_id="output").style(show_copy_button=True)
+            followupquestion = gr.Textbox(
+                label="FOLLOW UP QUESTION", lines=2, elem_id="question")
+            followupButton = gr.Button(
+                value="FOLLOW UP", elem_id="buttonfollowup")
 
     submit_button.click(
-        handleSubmit, [brain_name, enhancer_name, question, temperature, maxTokens], output_text)
-
+        handleSubmit, [brain_name, enhancer_name, question, temperature, maxTokens], [output_text, memory])
+    followupButton.click(
+        runAgent, [memory, followupquestion, temperature, maxTokens], [output_text, memory])
     block_demo.load(getBrains, [], brain_name)
 
 block_demo.launch(show_api=False)