update user interface

app.py

@@ -13,7 +13,7 @@ import google.generativeai as genai
 
 
 # Set up LLM APIs
-llm_api_options = ['gemini-pro', 'gemini-1.5-flash', 'gpt-3.5-turbo-1106']
+llm_api_options = ['gemini-pro', 'gemini-1.5-flash', 'gpt-3.5-turbo-1106', 'gpt-4o-2024-05-13', 'gpt-4o-mini-2024-07-18']
 
 def query_gpt_model(
     prompt: str,
@@ -59,9 +59,14 @@ def query_gemini_model(
 ) -> str:
   del client
   model = genai.GenerativeModel(llm)
+  generation_config={'temperature': 0.0}
   while True and retries > 0:
     try:
-      response = model.generate_content(prompt, safety_settings=safety_settings)
+      response = model.generate_content(
+        prompt,
+        safety_settings=safety_settings,
+        generation_config=generation_config
+      )
       text_response = response.text.replace("**", "")
       return text_response
     except Exception as e:
@@ -118,6 +123,8 @@ with open('QuALITY.v1.0.1.htmlstripped.dev', 'r') as f:
 
     quality_dev.append(fields)
 
+# likely to succeed
+index_map = {'A': 1, 'B': 9, 'C': 13, 'D': 200}
 
 
 # Helper functions
@@ -419,7 +426,7 @@ def query_model_with_quality(
      # api_key = os.environ.get('GEMINI_API_KEY')
      genai.configure(api_key=api_key)
 
-   example = quality_dev[index]
+   example = quality_dev[index_map[index]]
    article = f"[Title: {example['title']}]\n\n{example['article']}"
    pages, pagination = quality_pagination(example, model_name, client)
    print('Finish Pagination.')
@@ -446,7 +453,7 @@ with gr.Blocks() as demo:
     We implement ReadAgent as a simple prompting system that uses the advanced language capabilities of LLMs to (1) decide what content to store together in a memory episode (**Episode Pagination**), (2) compress those memory episodes into short episodic memories called gist memories (**Memory Gisting**), and (3) take actions to look up passages in the original text if ReadAgent needs to remind itself of relevant details to complete a task (**Parallel Lookup and QA**)
     This demo can handle long-document reading comprehension tasks ([QuALITY](https://arxiv.org/abs/2112.08608); max 6,000 words) efficiently.
 
-    To get started, you can choose an index of QuALITY dataset.
+    To get started, you can choose an example article from QuALITY dataset.
     This demo uses Gemini API or OpenAI API so it requires the corresponding API key.
     """)
     with gr.Row():
@@ -457,9 +464,27 @@ with gr.Blocks() as demo:
               lines=1,
               type="password",
             )
-            index = gr.Dropdown(list(range(len(quality_dev))), value=13, label="QuALITY Index")
+            # index = gr.Dropdown(list(range(len(quality_dev))), value=13, label="QuALITY Index")
+            index = gr.Radio(['A', 'B', 'C', 'D'], label="Example Article", value='A')
+            with gr.Row():
+                example_article_a = gr.Textbox(
+                  label="Example Article (A)",
+                  lines=10,
+                  value=f"[Title: {quality_dev[index_map['A']]['title']}]\n\n{quality_dev[index_map['A']]['article']}")
+                example_article_a = gr.Textbox(
+                  label="Example Article (B)",
+                  lines=10,
+                  value=f"[Title: {quality_dev[index_map['B']]['title']}]\n\n{quality_dev[index_map['B']]['article']}")
+                example_article_a = gr.Textbox(
+                  label="Example Article (C)",
+                  lines=10,
+                  value=f"[Title: {quality_dev[index_map['C']]['title']}]\n\n{quality_dev[index_map['C']]['article']}")
+                example_article_a = gr.Textbox(
+                  label="Example Article (D)",
+                  lines=10,
+                  value=f"[Title: {quality_dev[index_map['D']]['title']}]\n\n{quality_dev[index_map['D']]['article']}")
             button = gr.Button("Execute")
-            original_article = gr.Textbox(label="Original Article", lines=20)
+            choosen_article = gr.Textbox(label="Choosen Original Article", lines=20)
             # prompt_pagination = gr.Textbox(label="Episode Pagination Prompt Template", lines=5)
             pagination_results = gr.Textbox(label="(1) Episode Pagination", lines=20)
             # prompt_gisting = gr.Textbox(label="Memory Gisting Prompt Template", lines=5)
@@ -478,7 +503,7 @@ with gr.Blocks() as demo:
           # prompt_pagination, pagination_results,
           # prompt_gisting, gisting_results,
           # prompt_lookup, lookup_qa_results,
-          original_article,
+          choosen_article,
           pagination_results,
           gisting_results,
           lookup_qa_results,