Update app.py

app.py

@@ -6,6 +6,7 @@ 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 Qwen and Claude 3.5 Sonnet AI model options
 - Added duration sum display in Exercise Data tab
 - Shows total duration units (16th notes) for verification
 - Added DeepSeek AI model option
@@ -55,14 +56,18 @@ import os
 import subprocess as sp
 import base64
 import shutil
-from openai import OpenAI  # For DeepSeek API
+from openai import OpenAI  # For API models
 
 # -----------------------------------------------------------------------------
 # 3. Configuration & constants
 # -----------------------------------------------------------------------------
 MISTRAL_API_URL = "https://api.mistral.ai/v1/chat/completions"
 MISTRAL_API_KEY = "yQdfM8MLbX9uhInQ7id4iUTwN4h4pDLX"  # Replace with your key
-DEEPSEEK_API_KEY = "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"
+OPENROUTER_API_KEYS = {
+    "DeepSeek": "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",
+    "Qwen": "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",
+    "Claude": "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"
+}
 
 SOUNDFONT_URLS = {
     "Trumpet": "https://github.com/FluidSynth/fluidsynth/raw/master/sf2/Trumpet.sf2",
@@ -261,7 +266,7 @@ def get_technique_based_on_level(level: str) -> str:
     return random.choice(techniques.get(level, ["with slurs"]))
 
 # -----------------------------------------------------------------------------
-# 9. LLM Query Function (supports Mistral and DeepSeek)
+# 9. LLM Query Function (supports multiple models)
 # -----------------------------------------------------------------------------
 def query_llm(model_name: str, prompt: str, instrument: str, level: str, key: str,
               time_sig: str, measures: int) -> str:
@@ -321,19 +326,25 @@ 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 == "DeepSeek":
+    elif model_name in ["DeepSeek", "Qwen", "Claude"]:
         try:
             client = OpenAI(
                 base_url="https://openrouter.ai/api/v1",
-                api_key=DEEPSEEK_API_KEY,
+                api_key=OPENROUTER_API_KEYS[model_name],
             )
             
+            model_map = {
+                "DeepSeek": "deepseek/deepseek-chat-v3-0324:free",
+                "Qwen": "qwen/qwen3-14b:free",
+                "Claude": "anthropic/claude-3.5-sonnet:beta"
+            }
+            
             completion = client.chat.completions.create(
                 extra_headers={
                     "HTTP-Referer": "https://github.com/AdaptiveMusicExerciseGenerator",
                     "X-Title": "Music Exercise Generator",
                 },
-                model="deepseek/deepseek-chat-v3-0324:free",
+                model=model_map[model_name],
                 messages=[
                     {"role": "system", "content": system_prompt},
                     {"role": "user", "content": user_prompt},
@@ -347,7 +358,7 @@ def query_llm(model_name: str, prompt: str, instrument: str, level: str, key: st
             content = completion.choices[0].message.content
             return content.replace("```json","").replace("```","").strip()
         except Exception as e:
-            print(f"Error querying DeepSeek API: {e}")
+            print(f"Error querying {model_name} API: {e}")
             return get_fallback_exercise(instrument, level, key, time_sig, measures)
     
     else:
@@ -399,7 +410,7 @@ def generate_exercise(instrument: str, level: str, key: str, tempo: int, time_si
         return f"Error: {str(e)}", None, str(tempo), None, "0", time_signature, 0
 
 # -----------------------------------------------------------------------------
-# 12. Simple AI chat assistant (optional, shares LLM)
+# 12. AI chat assistant
 # -----------------------------------------------------------------------------
 def handle_chat(message: str, history: List, instrument: str, level: str, ai_model: str):
     if not message.strip():
@@ -423,19 +434,25 @@ 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 == "DeepSeek":
+    elif ai_model in ["DeepSeek", "Qwen", "Claude"]:
         try:
             client = OpenAI(
                 base_url="https://openrouter.ai/api/v1",
-                api_key=DEEPSEEK_API_KEY,
+                api_key=OPENROUTER_API_KEYS[ai_model],
             )
             
+            model_map = {
+                "DeepSeek": "deepseek/deepseek-chat-v3-0324:free",
+                "Qwen": "qwen/qwen3-14b:free",
+                "Claude": "anthropic/claude-3.5-sonnet:beta"
+            }
+            
             completion = client.chat.completions.create(
                 extra_headers={
                     "HTTP-Referer": "https://github.com/AdaptiveMusicExerciseGenerator",
                     "X-Title": "Music Exercise Generator",
                 },
-                model="deepseek/deepseek-chat-v3-0324:free",
+                model=model_map[ai_model],
                 messages=messages,
                 temperature=0.7,
                 max_tokens=500,
@@ -452,7 +469,7 @@ def handle_chat(message: str, history: List, instrument: str, level: str, ai_mod
         return "", history
 
 # -----------------------------------------------------------------------------
-# 13. Gradio user interface definition (for humans!)
+# 13. Gradio user interface definition
 # -----------------------------------------------------------------------------
 def create_ui() -> gr.Blocks:
     with gr.Blocks(title="Adaptive Music Exercise Generator", theme="soft") as demo:
@@ -466,7 +483,7 @@ def create_ui() -> gr.Blocks:
                 with gr.Group(visible=True) as params_group:
                     gr.Markdown("### Exercise Parameters")
                     ai_model = gr.Radio(
-                        ["Mistral", "DeepSeek"],
+                        ["Mistral", "DeepSeek", "Qwen", "Claude"],
                         value="Mistral",
                         label="AI Model"
                     )