Update app.py

app.py

@@ -5,6 +5,7 @@ from config import file_id_htl_biotech, file_id_kamera_express, file_id_smart_sd
 import gdown
 import pandas as pd
 from hogwarts import get_answer
+from hogwats_gemini import get_answer as get_answer_gemini
 from evaluator import eval_answer
 import nltk
 nltk.download('punkt')
@@ -119,6 +120,25 @@ Score of **{eval_score["score"]}/5**
 
 Rationale:
 
+{eval_score["rationale_based_on_scoring_rules"]}
+"""
+    
+    return response, eval_md
+
+def generate_chat_answer_gemini(company_name, query):
+    
+    df = dico[name_to_id_mapper[company_name]]["data"]
+    content = df["content"].values
+    response = get_answer_gemini(query, company_name, content)
+    print("=====> Evaluating answer quality...")
+    eval_score = eval(eval_answer(query, response))
+    eval_md = f"""
+### Evalation of how well the response answer the intial question
+
+Score of **{eval_score["score"]}/5**
+
+Rationale:
+
 {eval_score["rationale_based_on_scoring_rules"]}
 """
     
@@ -143,11 +163,14 @@ with gr.Blocks(title=title,theme='nota-ai/theme') as demo:
                 with gr.Accordion("Accuracy score", open=False):
                     evaluator = gr.Markdown("Waiting for answer to evaluate...")
                 chat_output = gr.Markdown("Waiting for question...")
-            with gr.Tab("Chat - ICL", interactive=False):
-                with gr.Column(scale=5):
-                    chat_input_gemini = gr.Textbox(placeholder="Chat input", lines=2, label="Retrieve anything from the dataroom")
-                with gr.Column(scale=1):
-                    chat_submit_button_gemini = gr.Button(value="Submit")
+            with gr.Tab("Chat - ICL", interactive=True):
+                with gr.Row():
+                    with gr.Column(scale=5):
+                        chat_input_gemini = gr.Textbox(placeholder="Chat input", lines=2, label="Retrieve anything from the dataroom")
+                    with gr.Column(scale=1):
+                        chat_submit_button_gemini = gr.Button(value="Submit")
+                with gr.Accordion("Accuracy score", open=False):
+                    evaluator_gemini = gr.Markdown("Waiting for answer to evaluate...")
                 chat_output_gemini = gr.Markdown("Waiting for question...")
             with gr.Tab("Data", interactive = True):
                 with gr.Tab("Sunday Naturals"):
@@ -191,7 +214,7 @@ with gr.Blocks(title=title,theme='nota-ai/theme') as demo:
                     
     fn = partial(get_data_room_overview)
     fn_chat = partial(generate_chat_answer)
-    # fn_chat_gemini = partial(generate_chat_answer_gemini)
+    fn_chat_gemini = partial(generate_chat_answer_gemini)
     submit_button.click(fn=fn, inputs=[company_name], outputs=[
         data_room_overview,
         sunday_naturals_web,
@@ -209,7 +232,7 @@ with gr.Blocks(title=title,theme='nota-ai/theme') as demo:
     
     
     chat_submit_button.click(fn=fn_chat, inputs=[company_name, chat_input], outputs=[chat_output, evaluator])
-    # chat_submit_button_gemini.click(fn=fn_chat, inputs=[company_name, chat_input_gemini], outputs=[chat_output_gemini])
+    chat_submit_button_gemini.click(fn=fn_chat_gemini, inputs=[company_name, chat_input_gemini], outputs=[chat_output_gemini, evaluator_gemini])
 login = os.environ.get("login")
 pwd = os.environ.get("pwd")