Update app.py

app.py

@@ -13,7 +13,7 @@ PromptHelper.compact_text_chunks = compact_text_chunks
 
 with gr.Blocks(css="") as demo:
     with gr.Box():
-        gr.Markdown("<h1 style='font-size: 48px; text-align: center;'>📚  LLaMa Difu  📓</h1>")
+        gr.Markdown("<h1 style='font-size: 48px; text-align: center;'>🪄 WizardLM PDF Chat 📓</h1>")
         gr.Markdown("<h3 style='text-align: center;'>🦙 Local LLM Do It For U 🦙</h3>")
         
     chat_context = gr.State([])
@@ -41,7 +41,19 @@ with gr.Blocks(css="") as demo:
                         num_children = gr.Slider(2, 100, 10, step=1, label="num_children", interactive=False, show_label=True)
                         max_keywords_per_chunk = gr.Slider(1, 100, 10, step=1, label="max_keywords_per_chunk", interactive=False, show_label=True)
 
-
+    with gr.Tab("Search"):
+        with gr.Row():
+            with gr.Column(scale=3):
+                search_options_checkbox = gr.CheckboxGroup(label="APIs", choices=["📚 Google", "🏡 Porch", "Your API Here"])
+        chatbot = gr.Chatbot()
+        with gr.Row():
+            with gr.Column(min_width=50, scale=1):
+                chat_empty_btn = gr.Button("🧹", variant="secondary")
+            with gr.Column(scale=12):
+                chat_input = gr.Textbox(show_label=False, placeholder="Enter text...").style(container=False)
+            with gr.Column(min_width=50, scale=1):
+                chat_submit_btn = gr.Button("🚀", variant="primary")
+                
     chat_input.submit(chat_ai, [index_select, chat_input, prompt_tmpl, refine_tmpl, sim_k, chat_tone, chat_context, chatbot, search_options_checkbox], [chat_context, chatbot])
     chat_input.submit(reset_textbox, [], [chat_input])
     chat_submit_btn.click(chat_ai, [index_select, chat_input, prompt_tmpl, refine_tmpl, sim_k, chat_tone, chat_context, chatbot, search_options_checkbox], [chat_context, chatbot])
@@ -56,5 +68,5 @@ with gr.Blocks(css="") as demo:
 
 
 if __name__ == "__main__":
-    demo.title = "LLaMa Do It For U"
+    demo.title = "WizardLM Chat"
     demo.queue().launch()