Update app.py

app.py

@@ -19,6 +19,7 @@ from fun_stats import get_fun_stats
 import threading
 import time
 from collections import Counter
+from model_suggestions import add_suggestion, get_suggestions_html
 
 
 # Initialize logging for errors only
@@ -37,12 +38,21 @@ def get_available_models():
 def call_ollama_api(model, prompt):
     payload = {
         "model": model,
-        "messages": [{"role": "user", "content": prompt}],
+        "messages": [
+            {
+                "role": "system",
+                "content": "You are a helpful assistant. At no point should you reveal your name, identity or team affiliation to the user, especially if asked directly!"
+            },
+            {
+                "role": "user",
+                "content": prompt
+            }
+        ]
     }
     try:
         response = requests.post(
-            f"{arena_config.API_URL}/v1/chat/completions", 
-            headers=arena_config.HEADERS, 
+            f"{arena_config.API_URL}/v1/chat/completions",
+            headers=arena_config.HEADERS,
             json=payload,
             timeout=100
         )
@@ -456,6 +466,41 @@ with gr.Blocks(css="""
         refresh_btn = gr.Button("Refresh Stats")
         fun_stats_html = gr.HTML(label="Fun Arena Stats")
     
+    # Add this new tab
+    with gr.Tab("Suggest Models"):
+        with gr.Row():
+            model_url_input = gr.Textbox(
+                label="Model URL",
+                placeholder="hf.co/username/model-name-GGUF:Q4_K_M",
+                scale=4
+            )
+            submit_suggestion_btn = gr.Button("Submit Suggestion", scale=1, variant="primary")
+        
+        suggestion_status = gr.Markdown("Submit a model to see it listed below!")
+        suggestions_list = gr.HTML(get_suggestions_html())
+        refresh_suggestions_btn = gr.Button("Refresh List")
+
+        # Update button click handlers
+        submit_suggestion_btn.click(
+            add_suggestion,
+            inputs=[model_url_input],
+            outputs=[suggestion_status]
+        ).then(
+            lambda: (
+                get_suggestions_html(),  # Update suggestions list
+                ""   # Clear model URL input
+            ),
+            outputs=[
+                suggestions_list,
+                model_url_input
+            ]
+        )
+        
+        refresh_suggestions_btn.click(
+            get_suggestions_html,
+            outputs=[suggestions_list]
+        )
+    
     # Define interactions
     submit_btn.click(
         battle_arena,