added UI components

app.py

@@ -161,7 +161,7 @@ def radio_on_change(
     )
 
 
-title = """<h1 align="center">🔥Large Language Model API 🚀Streaming🚀</h1>"""
+title = """<h1 align="center">🔥Teaching Assistant Chatbot 🚀Streaming🚀</h1>"""
 description = """Language models can be conditioned to act like dialogue agents through a conversational prompt that typically takes the form:
 
 ```
@@ -184,31 +184,62 @@ with gr.Blocks(
                 #chatbot {height: 520px; overflow: auto;}"""
 ) as demo:
     gr.HTML(title)
-    with gr.Column(elem_id="col_container"):
+    with gr.Row():
         model = gr.Radio(
             value="OpenAssistant/oasst-sft-1-pythia-12b",
-            choices=[
-                "OpenAssistant/oasst-sft-1-pythia-12b",
-                # "togethercomputer/GPT-NeoXT-Chat-Base-20B",
-                "Rallio67/joi2_20Be_instruct_alpha",
-                "google/flan-t5-xxl",
-                "google/flan-ul2",
-                "bigscience/bloom",
-                "bigscience/bloomz",
-                "EleutherAI/gpt-neox-20b",
-            ],
-            label="Model",
-            interactive=True,
-        )
-
-        chatbot = gr.Chatbot(elem_id="chatbot")
-        inputs = gr.Textbox(
-            placeholder="Hi there!", label="Type an input and press Enter"
-        )
-        disclaimer = gr.Markdown(openchat_disclaimer, visible=False)
-        state = gr.State([])
-        b1 = gr.Button()
-
+                choices=[
+                    "OpenAssistant/oasst-sft-1-pythia-12b",
+                    # "togethercomputer/GPT-NeoXT-Chat-Base-20B",
+                    "Rallio67/joi2_20Be_instruct_alpha",
+                    "google/flan-t5-xxl",
+                    "google/flan-ul2",
+                    "bigscience/bloom",
+                    "bigscience/bloomz",
+                    "EleutherAI/gpt-neox-20b",
+                 ],
+                label="Model",
+                interactive=True,
+        )   
+    with gr.Row():
+        with gr.Column():    
+            use_gpt3_checkbox = gr.Checkbox(label="Include GPT-3 (paid)?")
+        with gr.Column():
+            use_equation_checkbox = gr.Checkbox(label="Prioritize equations?")
+            
+    with gr.Row():
+        with gr.Column():
+            chatbot = gr.Chatbot(elem_id="chatbot")
+            inputs = gr.Textbox(
+                placeholder="Ask an Electrical Engineering question!", label="Type an input and press Enter"
+            )
+            examples = gr.Examples(
+                examples=[
+                    ["What is a Finite State Machine?"],
+                    ["How do you design a functional a Two-Bit Gray Code Counter?"],
+                ],
+                inputs=[inputs],  
+                outputs=[],
+            ) 
+    gr.Markdown("The top 3 retrieved contexts are:")
+    with gr.Row():
+        with gr.Column():
+            # add context retrieval here
+            context1 = gr.Textbox(label="Context 1")
+        with gr.Column():
+            context2 = gr.Textbox(label="Context 2")
+        with gr.Column():
+            context3 = gr.Textbox(label="Context 3")
+    
+    gr.Markdown("Lecture Slides:")
+    with gr.Row():
+        # add CLIP part here
+        lec_gallery = gr.Gallery(label="Lecture images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
+            
+    disclaimer = gr.Markdown(openchat_disclaimer, visible=False)
+    state = gr.State([])
+    run_btn = gr.Button(variant='primary',)
+
+    with gr.Row():
         with gr.Accordion("Parameters", open=False):
             typical_p = gr.Slider(
                 minimum=-0,
@@ -256,6 +287,7 @@ with gr.Blocks(
             )
             watermark = gr.Checkbox(value=False, label="Text watermarking")
 
+
     model.change(
         lambda value: radio_on_change(
             value,
@@ -295,7 +327,7 @@ with gr.Blocks(
         ],
         [chatbot, state],
     )
-    b1.click(
+    run_btn.click(
         predict,
         [
             model,
@@ -311,7 +343,7 @@ with gr.Blocks(
         ],
         [chatbot, state],
     )
-    b1.click(reset_textbox, [], [inputs])
+    run_btn.click(reset_textbox, [], [inputs])
     inputs.submit(reset_textbox, [], [inputs])
 
     gr.Markdown(description)