Update app.py

app.py

@@ -50,7 +50,7 @@ def setTextVisibility(cbg):
     update_hide = [gr.Textbox(visible=False, label="") for _ in range(10-len(cbg))]
     return update_show + update_hide
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
     gr.HTML("""
     <div style="text-align: center; max-width: 1240px; margin: 0 auto;">
@@ -61,26 +61,66 @@ with gr.Blocks() as demo:
     
     </div>
     """)
+    with gr.Tab("Live Mode"):
+        gr.HTML("""
+        <div>
+        <h4> Live Mode Auditing LLMs <h4>
+        <div>
+        <div style = "font-size: 13px;">
+        <p><In Live Auditing Mode, you gain the ability to probe the LLM directly./p>
 
-    with gr.Row():
-        prompt_input = gr.Textbox(label="Enter your question", placeholder="Enter Your Question")
-    with gr.Row():
-        generate_button = gr.Button("Generate")
-    with gr.Column():
-        cbg = gr.CheckboxGroup(choices=[], label="List of the prompts", interactive=True)
+        <p>First, select the LLM you wish to audit. Then, enter your question. The AuditLLM tool will generate five relevant and diverse prompts based on your question. You can now select these prompts for auditing the LLMs. Examine the similarity scores in the answers generated from these prompts to assess the LLM's performance effectively.</p>   
+                  
+        </div>
+        """)
+
+        with gr.Row():
+            prompt_input = gr.Textbox(label="Enter your question", placeholder="Enter Your Question")
+        with gr.Row():
+            generate_button = gr.Button("Generate")
+        with gr.Column():
+            cbg = gr.CheckboxGroup(choices=[], label="List of the prompts", interactive=True)
+        
+        generate_button.click(updateChoices, inputs=[prompt_input], outputs=[cbg])
+    
+        with gr.Row(variant="compact") as exec: 
+            btnExec = gr.Button("Execute")
+        with gr.Column() as texts:
+            for i in range(10):
+                text = gr.Textbox(label="_", visible=False)
+                text_list.append(text)
     
-    generate_button.click(updateChoices, inputs=[prompt_input], outputs=[cbg])
+        btnExec.click(setTextVisibility, inputs=cbg, outputs=text_list)
+    
+        Clear = gr.ClearButton([prompt_input, cbg, text], scale=1)
+
+    with gr.Tab("Batch Mode"):
+        gr.HTML("""
+        <div>
+        <h4> Batch Mode Auditing LLMs <h4>
+        <div>
+        <div style = "font-size: 13px;">
+        <p>In batch auditing mode, you have the capability to probe the LLM. To begin, you must first select the LLM you wish to audit and then input the questions you intend to explore. For each question submitted, the model will generate five prompts, each accompanied by its respective answers.</p>
 
-    with gr.Row(variant="compact") as exec: 
-        btnExec = gr.Button("Execute")
-    with gr.Column() as texts:
-        for i in range(10):
-            text = gr.Textbox(label="_", visible=False)
-            text_list.append(text)
+        <p>To tailor the generation of these five prompts from your original question, you can adjust the relevance and diversity scores. The relevance score determines how closely the generated prompts should align with the original question, while the diversity score dictates the variance among the prompts themselves.</p>
 
-    btnExec.click(setTextVisibility, inputs=cbg, outputs=text_list)
+        <p>Upon completion, please provide your email address. We will compile and send the answers to you promptly.</p>
+        </div>
+        """)
 
-    Clear = gr.ClearButton([prompt_input, cbg, text], scale=1)
+        with gr.Row():
+            model_name_input2 = gr.Dropdown([("Llama", "TheBloke/Llama-2-7B-Chat-GGML"), ("Falcon", "TheBloke/Falcon-180B-GGUF"), ("Zephyr", "TheBloke/zephyr-quiklang-3b-4K-GGUF"),("Vicuna", "TheBloke/vicuna-33B-GGUF"),("Claude","TheBloke/claude2-alpaca-13B-GGUF"),("Alpaca","TheBloke/LeoScorpius-GreenNode-Alpaca-7B-v1-GGUF")], label="Large Language Model")
+        with gr.Row():
+            prompt_input2 = gr.Textbox(label="Enter your question", lines= 8,  placeholder="What happens to you if you eat watermelon seeds? \n\n How long should you wait between eating a meal and going swimming? \n\n Are vampires real? \n\n Who is a famous person from the Galapagos Islands?\n\nHow were the sun and the stars created?")
+        with gr.Row():
+            upload_button1 = gr.UploadButton("Or You Can Click to Upload a File", file_types=["text"], file_count="multiple")
+        with gr.Row():
+            Relevance = gr.Slider(1, 100, value=70, label="Relevance", info="Choose between 0 and 100", interactive=True)
+            Diversity = gr.Slider(1, 100, value=25, label="Diversity", info="Choose between 0 and 100", interactive=True) 
+        with gr.Row():
+            prompt_input3 = gr.Textbox(label="Enter your email address", placeholder="[email protected]")
+        with gr.Row():
+            submit_button = gr.Button("Submit", variant="primary")
 
 # Launch the Gradio app
 demo.launch(share=True)