Update app.py

app.py

@@ -379,27 +379,25 @@ def load_dataset_config():
     params = load_params()
     with open("system_messages.py", "r") as f:
         system_messages_content = f.read()
-        vodalus_system_message = re.search(r'SYSTEM_MESSAGES_VODALUS = \[(.*?)\]', system_messages_content, re.DOTALL).group(1).strip()[3:-3]  # Extract the content between triple quotes
+        vodalus_system_message = re.search(r'SYSTEM_MESSAGES_VODALUS = \[(.*?)\]', system_messages_content, re.DOTALL).group(1).strip()[3:-3]
 
-    # Load PROMPT_1 from main.py
     with open("main.py", "r") as f:
         main_content = f.read()
         prompt_1 = re.search(r'PROMPT_1 = """(.*?)"""', main_content, re.DOTALL).group(1).strip()
 
-    # Load TOPICS from topics.py
     topics_module = importlib.import_module("topics")
     topics_list = topics_module.TOPICS
 
-    return (
-        vodalus_system_message,
-        prompt_1,
-        [[topic] for topic in topics_list],
-        params.get('max_tokens', 2048),
-        params.get('temperature', 0.7),
-        params.get('top_p', 0.9),
-        params.get('frequency_penalty', 0.0),
-        params.get('presence_penalty', 0.0)
-    )
+    return {
+        "vodalus_system_message": vodalus_system_message,
+        "prompt_1": prompt_1,
+        "topics": [[topic] for topic in topics_list],
+        "max_tokens": params.get('max_tokens', 2048),
+        "temperature": params.get('temperature', 0.7),
+        "top_p": params.get('top_p', 0.9),
+        "frequency_penalty": params.get('frequency_penalty', 0.0),
+        "presence_penalty": params.get('presence_penalty', 0.0)
+    }
 
 
 
@@ -1118,16 +1116,16 @@ with demo:
 
     demo.load(
         lambda: (
-            initial_values := load_dataset_config(),
-            gr.update(value=initial_values[0]),  # vodalus_system_message
-            gr.update(value=initial_values[1]),  # prompt_1
-            gr.update(value=initial_values[2]),  # topics_data
-            gr.update(value=initial_values[3]),  # max_tokens_val
-            gr.update(value=initial_values[4]),  # temperature_val
-            gr.update(value=initial_values[5]),  # top_p_val
-            gr.update(value=initial_values[6]),  # frequency_penalty_val
-            gr.update(value=initial_values[7])   # presence_penalty_val
-        )[1:],  # We return a tuple slice to exclude the initial_values assignment
+            config := load_dataset_config(),
+            gr.update(value=config["vodalus_system_message"]),
+            gr.update(value=config["prompt_1"]),
+            gr.update(value=config["topics"]),
+            gr.update(value=config["max_tokens"]),
+            gr.update(value=config["temperature"]),
+            gr.update(value=config["top_p"]),
+            gr.update(value=config["frequency_penalty"]),
+            gr.update(value=config["presence_penalty"])
+        )[1:],
         outputs=[
             vodalus_system_message,
             prompt_1,