Update app.py

app.py

@@ -12,6 +12,7 @@ HF_API_KEY = os.getenv("HF_API_KEY")
 
 # Model endpoints configuration
 MODEL_ENDPOINTS = {
+    "DeepSeek-R1": "https://api-inference.huggingface.co/models/deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
     "Qwen2.5-72B-Instruct": "https://api-inference.huggingface.co/models/Qwen/Qwen2.5-72B-Instruct",
     "Llama3.3-70B-Instruct": "https://api-inference.huggingface.co/models/meta-llama/Llama-3.3-70B-Instruct",
     "Qwen2.5-Coder-32B-Instruct": "https://api-inference.huggingface.co/models/Qwen/Qwen2.5-Coder-32B-Instruct",
@@ -26,6 +27,10 @@ def query_model(model_name: str, messages: List[Dict[str, str]]) -> Generator[st
     
     # Model-specific prompt formatting with full history
     model_prompts = {
+        "DeepSeek-R1": (
+            f"<|im_start|>system\nProvide foundational analysis of:\n{conversation}<|im_end|>\n"
+            "<|im_start|>assistant\nFoundational perspective:"
+        ),
         "Qwen2.5-72B-Instruct": (
             f"<|im_start|>system\nCollaborate with other experts. Previous discussion:\n{conversation}<|im_end|>\n"
             "<|im_start|>assistant\nMy analysis:"
@@ -82,8 +87,8 @@ def respond(message: str, history: List[List[str]], session_id: str) -> Generato
     })
 
     # Model responses
-    model_names = ["Qwen2.5-Coder-32B-Instruct", "Qwen2.5-72B-Instruct", "Llama3.3-70B-Instruct"]
-    model_colors = ["🔵", "🟣", "🟡"]
+    model_names = ["DeepSeek-R1", "Qwen2.5-Coder-32B-Instruct", "Qwen2.5-72B-Instruct", "Llama3.3-70B-Instruct"]
+    model_colors = ["🔴", "🔵", "🟣", "🟡"]
     responses = {}
 
     # Initialize responses