Update app.py

app.py

@@ -45,10 +45,37 @@ def updateChoices(prompt):
     newChoices = generate_prompts(prompt)
     return gr.CheckboxGroup(choices=newChoices)
 
-def setTextVisibility(cbg):
-    update_show = [gr.Textbox(visible=True, label=text, value=answer_question(text)) for text in cbg]
-    update_hide = [gr.Textbox(visible=False, label="") for _ in range(10-len(cbg))]
-    return update_show + update_hide
+def setTextVisibility(cbg, model_name_input):
+    sentences = []
+    result = []
+    model = SentenceTransformer('all-mpnet-base-v2')
+    exclude_words = {"a", "the", "for", "from", "of", "in", "over", "as", "on", "is", "am", "have", "an", "has", "had", "and", "by", "it", "its", "those", "these", "above", "to", "However"}
+    sentences_org = ["In a quaint little town nestled in the heart of the mountains, a small bakery famous for its artisanal breads and pastries had a line of customers stretching out the door, eagerly waiting to savor the freshly baked goods that were known far and wide for their delightful flavors.",
+                "Within a picturesque mountain village, there stood a renowned bakery, celebrated for its handcrafted bread and sweet treats, attracting a long queue of patrons each morning, all keen to enjoy the baked delicacies that had gained widespread acclaim for their exceptional taste.",
+                "A charming bakery, located in a small mountainous hamlet, renowned for producing exquisite handmade pastries and bread, was bustling with a crowd of eager customers lined up outside, each anticipating the chance to indulge in the famous baked items celebrated for their extraordinary deliciousness.",
+                "In a cozy, mountain-encircled village, a beloved bakery was the center of attraction, known for its traditional baking methods and delightful pastries, drawing a consistent stream of people waiting outside, all desiring to experience the renowned flavors that made the bakery's products distinctively mouth-watering."]
+    for text in cbg:
+         sentences.append(answer_question(text, model_name_input))
+
+    highlighted_sentences = []
+    for i, sentence in enumerate(sentences):
+        other_sentences = sentences[:i] + sentences[i+1:]
+        highlighted_sentence = highlight_words(sentence, other_sentences, model, exclude_words)
+        highlighted_sentences.append(highlighted_sentence)
+        
+    for idx, sentence in enumerate(highlighted_sentences):
+        result.append("<p><strong>"+ cbg[idx] +"</strong></p><p>"+ sentence +"</p><br/>")
+
+    score = round(calculate_similarity_score(sentences))
+
+    final_html = f"""<div>{result}<div style="text-align: center; font-size: 24px; font-weight: bold;">Similarity Score: {score}</div></div>"""
+
+
+    return final_html
+
+def upload_file(files):
+    file_paths = [file.name for file in files]
+    return file_paths
 
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
 
@@ -74,25 +101,36 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         </div>
         """)
 
+        with gr.Row():
+            model_name_input = 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_input = gr.Textbox(label="Enter your question", placeholder="Enter Your Question")
         with gr.Row():
-            generate_button = gr.Button("Generate")
+            generate_button = gr.Button("Generate", variant="primary", min_width=300)
         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.Row() as exec: 
+            btnExec = gr.Button("Execute", variant="primary", min_width=200)
+
+
         with gr.Column() as texts:
             for i in range(10):
                 text = gr.Textbox(label="_", visible=False)
                 text_list.append(text)
-    
-        btnExec.click(setTextVisibility, inputs=cbg, outputs=text_list)
-    
-        Clear = gr.ClearButton([prompt_input, cbg, text], scale=1)
+
+        with gr.Column():
+            html_result = gr.HTML("""<div style="color: red"></div>""")
+
+        #btnExec.click(setTextVisibility, inputs=[cbg, model_name_input], outputs=text_list)
+        btnExec.click(setTextVisibility, inputs=[cbg, model_name_input], outputs=html_result)
+        gr.HTML("""
+        <div style="text-align: center; font-size: 24px; font-weight: bold;">Similarity Score:</div>
+                """)
+
+        clear = gr.ClearButton(link = "http://127.0.0.1:7860") 
 
     with gr.Tab("Batch Mode"):
         gr.HTML("""