Functionality rework

app.py

@@ -4,6 +4,7 @@ import os
 import boto3
 import pandas as pd
 from copy import copy
+from random import choice
 
 import queue
 
@@ -17,6 +18,7 @@ TEST_MD = None
 CURRENT_MODELS = queue.LifoQueue()
 MODEL_LIB = {'TINYLLAMA': get_tinyllama, "QWEN2INS1B": get_qwen2ins1b, "RUBASE": GigaChat.get_giga}
 GEN_LIB = {'TINYLLAMA': response_tinyllama, "QWEN2INS1B": response_qwen2ins1b, "RUBASE": response_gigachat}
+MODEL_LIST = ["TINYLLAMA", "QWEN2INS1B", "RUBASE"]
 
 def model_gen(
         content,
@@ -35,7 +37,7 @@ def model_gen(
     if len(content) == 0:
         return '', []
     
-    chat_history = [chat_history[-1]] if no_context and len(chat_history)>=1  else chat_history
+    chat_history = [] if no_context else chat_history
     
     return model_response(
         content, 
@@ -88,13 +90,13 @@ def model_manager(
     INIT_MODELS[add_model] = model_lib[add_model]()
 
 def tab_online_arena():
-    global S3_SESSION, GEN_LIB, MODEL_LIB, INIT_MODELS, CURRENT_MODELS
+    global S3_SESSION, GEN_LIB, MODEL_LIB, INIT_MODELS, CURRENT_MODELS, MODEL_LIST
     with gradio.Row():
         with gradio.Column():
-            model_left = gradio.Dropdown(["TINYLLAMA", "QWEN2INS1B", "RUBASE"], value="QWEN2INS1B", interactive=True, multiselect=False, label="Left model")
+            model_left = gradio.Dropdown(MODEL_LIST, value=choice(MODEL_LIST), interactive=True, multiselect=False, label="Left model")
             chatbot_left = gradio.Chatbot()
         with gradio.Column():
-            model_right = gradio.Dropdown(["TINYLLAMA", "QWEN2INS1B", "RUBASE"], value= "RUBASE", interactive=True, multiselect=False, label="Right model")
+            model_right = gradio.Dropdown(MODEL_LIST, value=choice(MODEL_LIST), interactive=True, multiselect=False, label="Right model")
             chatbot_right = gradio.Chatbot()
 
     with gradio.Row():
@@ -197,13 +199,14 @@ def tab_leaderboard():
         with gradio.Blocks() as demo:
             gradio.DataFrame(df)
 
-def tab_offline_arena(): 
+def tab_offline_arena():
+    global MODEL_LIST
     # with gradio.Row(): 
     #     btn_show_history = gradio.Button("🎲  Click here to sample an example + a pair of LLM outputs! ", elem_classes="sample_button")
     with gradio.Row():
         with gradio.Column(scale=2):
             with gradio.Accordion("Choose models to sample from", open=False, elem_classes="accordion-label"):
-                model_options = list(MODEL_LIB.keys())  
+                model_options = MODEL_LIST
                 selected_models = gradio.CheckboxGroup(model_options, info="", value=model_options, show_label=False, elem_id="select-models") 
                 clear_button = gradio.Button("Clear", elem_classes="btn_boderline_gray", scale=1)
                 # clear the selected_models