Update app.py

app.py

@@ -16,37 +16,46 @@ def getBrains():
     return gr.update(choices=active_indexes)
 
 
+prompt_names = ["Research & Summarize"]
+prompt_enhancers = ["""Please act as a researcher and summarizer for the [TOPIC] I list at the end of this prompt. Create a concise, accurate, and objective summary of the main points and findings. Use clear headlines, bullet points, and sub-bullet points where needed. Keep in mind the following guidelines:
+Avoid personal opinions or interpretations; focus on objectively presenting the information.
+Write the summary in your own words.
+Reference data sources when necessary.
+Ensure the summary is clear and concise.
+Please write in English language.
+"""]
+
+
 bg_color = "#c5dde0"
 s_color = "#1d2230"
 mycss = """
     .gradio-container {{background-color: {bgcolor}}}
     #title {{margin-top:33%;margin-bottom:25px;display:flex;justify-content:center;align-items:center}}
+    .gap.svelte-vt1mxs {{gap:0}}
     #title h1 {{font-weight:900;color:{scolor}}}
     #advanced {{font-weight:600;background-color:#ffffff}}
     #secondrow {{padding:0 6%;gap:30px}}    
-
-
     #name {{background-color: {bgcolor};border-style:none;border-width:0;box-shadow:none;padding-left:0;padding-right:0}}
     #name .svelte-1gfkn6j {{background-color:{bgcolor};color:{scolor};font-size:18px}}
-    
 
+    #enhancer-name {{background-color: {bgcolor};border-style:none;border-width:0;box-shadow:none;padding-left:0;padding-right:0}}
+    #enhancer-name .svelte-1gfkn6j {{background-color:{bgcolor};color:{scolor};font-size:18px}}
+    #enhancer-name .svelte-e8n7p6 {{color:{scolor};padding-left:8px}}
+    
     #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 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}}
 """
 formatted_css = mycss.format(bgcolor=bg_color, scolor=s_color)
 
 
-def handleSubmit(brain_name, question, temperature, maxTokens):
+def handleSubmit(brain_name, enhancer, question, temperature, maxTokens):
     print(brain_name)
     if (brain_name == "" and question == ""):
         return "Please select Brain Name & Enter Question"
@@ -54,6 +63,9 @@ def handleSubmit(brain_name, question, temperature, maxTokens):
         return "Please select Brain Name"
     if (question == ""):
         return "Please Enter Question"
+    if (enhancer != ""):
+        promptIndex = prompt_names.index(enhancer)
+        question = prompt_enhancers[promptIndex]+question
     return askQuestion(brain_name, question, temperature, maxTokens)
 
 
@@ -72,6 +84,8 @@ with gr.Blocks(theme=gr.themes.Soft(), css=formatted_css) as block_demo:
 
             brain_name = gr.Dropdown(
                 label="Brain Name", choices=None, elem_id="name", multiselect=False, interactive=True)
+            enhancer_name = gr.Dropdown(
+                label="Prompt Template (Optional)", choices=prompt_names, elem_id="enhancer-name", multiselect=False, interactive=True)
             question = gr.Textbox(
                 label="Question", lines=2, elem_id="question")
 
@@ -85,11 +99,11 @@ with gr.Blocks(theme=gr.themes.Soft(), css=formatted_css) as block_demo:
 
         with gr.Column(scale=1, elem_id="outputCol",):
             output_text = gr.TextArea(
-                label="Brain Output", lines=14, elem_id="output").style(show_copy_button=True)
+                label="Brain Output", lines=17, elem_id="output").style(show_copy_button=True)
 
     submit_button.click(
-        handleSubmit, [brain_name, question, temperature, maxTokens], output_text)
+        handleSubmit, [brain_name, enhancer_name, question, temperature, maxTokens], output_text)
 
     block_demo.load(getBrains, [], brain_name)
 
-block_demo.launch(show_api=False)
+block_demo.launch(show_api=False)