Update app.py

app.py

@@ -61,7 +61,9 @@ start_backup_thread()
 
 # Function to get available models (using predefined list)
 def get_available_models():
-    return [model[0] for model in config.get_approved_models()]
+    # Filter out models that have been archived
+    active_models = [model[0] for model in config.get_approved_models() if model[0] not in leaderboard.load_archived_models()]
+    return active_models
 
 # Function to get recent opponents for a model
 recent_opponents = {}
@@ -95,7 +97,8 @@ def call_ollama_api(model, prompt):
                     "content": prompt
                 }
             ],
-            timeout=180
+            timeout=180,
+            max_tokens=config.MAX_TOKENS
         )
         logger.info(f"Received response for model: {model}")
         
@@ -574,6 +577,20 @@ with gr.Blocks(css="""
             interactive=True,
             label="ELO Leaderboard"
         )
+
+    # Archived Models Tab
+    with gr.Tab("Archived Models"):
+        gr.Markdown("""
+        ### Archived Models
+        These models have reached the battle threshold and have been moved to the archive.
+        """)
+        archived_models_table = gr.Dataframe(
+            headers=["Model", "Wins", "Losses", "Total Battles", "Win Rate"],
+            row_count=10,
+            col_count=5,
+            interactive=True,
+            label="Archived Models"
+        )
     
     # Latest Updates Tab
     with gr.Tab("Latest Updates"):
@@ -626,6 +643,7 @@ with gr.Blocks(css="""
     # Update leaderboard on launch
     demo.load(get_leaderboard, outputs=leaderboard)
     demo.load(get_elo_leaderboard, outputs=elo_leaderboard)
+    demo.load(leaderboard.get_archived_models_list, outputs=archived_models_table)
 
 if __name__ == "__main__":
     # Initialize ELO ratings before launching the app