Update app.py

app.py

@@ -6,10 +6,10 @@ Generates custom musical exercises with LLM, perfectly fit to user-specified num
 AND time signature, guaranteeing exact durations in MIDI and in the UI!
 
 Major updates:
-- Added Gemma and Kimi Dev 72b AI model options
-- Added DeepSeek R1 model option
+- Added Gemma, Kimi Dev 72b, and Llama 3.1 AI model options
 - Added duration sum display in Exercise Data tab
 - Shows total duration units (16th notes) for verification
+- Added DeepSeek AI model option
 - Fixed difficulty level implementation
 - Maintained all original functionality
 """
@@ -64,10 +64,11 @@ from openai import OpenAI  # For API models
 MISTRAL_API_URL = "https://api.mistral.ai/v1/chat/completions"
 MISTRAL_API_KEY = "yQdfM8MLbX9uhInQ7id4iUTwN4h4pDLX"  # Replace with your key
 OPENROUTER_API_KEYS = {
+    "DeepSeek": "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",
+    "Claude": "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",
     "Gemma": "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",
     "Kimi": "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",
-    "Claude": "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",
-    "DeepSeek-R1": "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"
+    "Llama 3.1": "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"
 }
 
 SOUNDFONT_URLS = {
@@ -327,7 +328,7 @@ def query_llm(model_name: str, prompt: str, instrument: str, level: str, key: st
             print(f"Error querying Mistral API: {e}")
             return get_fallback_exercise(instrument, level, key, time_sig, measures)
     
-    elif model_name in ["Gemma", "Kimi", "Claude", "DeepSeek-R1"]:
+    elif model_name in ["DeepSeek", "Claude", "Gemma", "Kimi", "Llama 3.1"]:
         try:
             client = OpenAI(
                 base_url="https://openrouter.ai/api/v1",
@@ -335,10 +336,11 @@ def query_llm(model_name: str, prompt: str, instrument: str, level: str, key: st
             )
             
             model_map = {
+                "DeepSeek": "deepseek/deepseek-chat-v3-0324:free",
+                "Claude": "anthropic/claude-3.5-sonnet:beta",
                 "Gemma": "google/gemma-3n-e2b-it:free",
                 "Kimi": "moonshotai/kimi-dev-72b:free",
-                "Claude": "anthropic/claude-3.5-sonnet:beta",
-                "DeepSeek-R1": "deepseek/deepseek-r1-0528:free"
+                "Llama 3.1": "meta-llama/llama-3.1-405b-instruct:free"
             }
             
             completion = client.chat.completions.create(
@@ -436,7 +438,7 @@ def handle_chat(message: str, history: List, instrument: str, level: str, ai_mod
             history.append((message, f"Error: {str(e)}"))
             return "", history
     
-    elif ai_model in ["Gemma", "Kimi", "Claude", "DeepSeek-R1"]:
+    elif ai_model in ["DeepSeek", "Claude", "Gemma", "Kimi", "Llama 3.1"]:
         try:
             client = OpenAI(
                 base_url="https://openrouter.ai/api/v1",
@@ -444,10 +446,11 @@ def handle_chat(message: str, history: List, instrument: str, level: str, ai_mod
             )
             
             model_map = {
+                "DeepSeek": "deepseek/deepseek-chat-v3-0324:free",
+                "Claude": "anthropic/claude-3.5-sonnet:beta",
                 "Gemma": "google/gemma-3n-e2b-it:free",
                 "Kimi": "moonshotai/kimi-dev-72b:free",
-                "Claude": "anthropic/claude-3.5-sonnet:beta",
-                "DeepSeek-R1": "deepseek/deepseek-r1-0528:free"
+                "Llama 3.1": "meta-llama/llama-3.1-405b-instruct:free"
             }
             
             completion = client.chat.completions.create(
@@ -486,7 +489,7 @@ def create_ui() -> gr.Blocks:
                 with gr.Group(visible=True) as params_group:
                     gr.Markdown("### Exercise Parameters")
                     ai_model = gr.Radio(
-                        ["Mistral", "Gemma", "Kimi", "Claude", "DeepSeek-R1"],
+                        ["Mistral", "DeepSeek", "Claude", "Gemma", "Kimi", "Llama 3.1"],
                         value="Mistral",
                         label="AI Model"
                     )