Update app.py

app.py

@@ -6,8 +6,8 @@ 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 2b model option
-- Added Claude 3.5 Sonnet model option
+- Added Gemma and Kimi Dev 72b AI model options
+- Added DeepSeek R1 model option
 - Added duration sum display in Exercise Data tab
 - Shows total duration units (16th notes) for verification
 - Fixed difficulty level implementation
@@ -64,9 +64,10 @@ 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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",
     "Gemma": "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",
-    "Claude": "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"
+    "Kimi": "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",
+    "Claude": "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",
+    "DeepSeek-R1": "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"
 }
 
 SOUNDFONT_URLS = {
@@ -326,7 +327,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 ["DeepSeek", "Gemma", "Claude"]:
+    elif model_name in ["Gemma", "Kimi", "Claude", "DeepSeek-R1"]:
         try:
             client = OpenAI(
                 base_url="https://openrouter.ai/api/v1",
@@ -334,15 +335,12 @@ def query_llm(model_name: str, prompt: str, instrument: str, level: str, key: st
             )
             
             model_map = {
-                "DeepSeek": "deepseek/deepseek-chat-v3-0324:free",
                 "Gemma": "google/gemma-3n-e2b-it:free",
-                "Claude": "anthropic/claude-3.5-sonnet:beta"
+                "Kimi": "moonshotai/kimi-dev-72b:free",
+                "Claude": "anthropic/claude-3.5-sonnet:beta",
+                "DeepSeek-R1": "deepseek/deepseek-r1-0528:free"
             }
             
-            # Special handling for Gemma to ensure JSON output
-            if model_name == "Gemma":
-                user_prompt += " Output in valid JSON format only."
-            
             completion = client.chat.completions.create(
                 extra_headers={
                     "HTTP-Referer": "https://github.com/AdaptiveMusicExerciseGenerator",
@@ -358,17 +356,9 @@ def query_llm(model_name: str, prompt: str, instrument: str, level: str, key: st
                 top_p=0.95,
                 frequency_penalty=0.2,
                 presence_penalty=0.2,
-                response_format={"type": "json_object"} if model_name == "Gemma" else None
             )
             content = completion.choices[0].message.content
-            
-            # Clean up Gemma output
-            if model_name == "Gemma":
-                content = content.replace("```json", "").replace("```", "").strip()
-                # Remove any text before the first [
-                content = re.sub(r'^[^[]*', '', content)
-            
-            return content
+            return content.replace("```json","").replace("```","").strip()
         except Exception as e:
             print(f"Error querying {model_name} API: {e}")
             return get_fallback_exercise(instrument, level, key, time_sig, measures)
@@ -382,10 +372,6 @@ def query_llm(model_name: str, prompt: str, instrument: str, level: str, key: st
 def safe_parse_json(text: str) -> Optional[list]:
     try:
         text = text.replace("'", '"')
-        # Remove any text before the first [ and after the last ]
-        text = re.sub(r'^[^[]*', '', text)
-        text = re.sub(r'[^\]]*$', '', text)
-        # Extract JSON array
         match = re.search(r"\[(\s*\[.*?\]\s*,?)*\]", text, re.DOTALL)
         if match:
             return json.loads(match.group(0))
@@ -450,7 +436,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 ["DeepSeek", "Gemma", "Claude"]:
+    elif ai_model in ["Gemma", "Kimi", "Claude", "DeepSeek-R1"]:
         try:
             client = OpenAI(
                 base_url="https://openrouter.ai/api/v1",
@@ -458,9 +444,10 @@ def handle_chat(message: str, history: List, instrument: str, level: str, ai_mod
             )
             
             model_map = {
-                "DeepSeek": "deepseek/deepseek-chat-v3-0324:free",
                 "Gemma": "google/gemma-3n-e2b-it:free",
-                "Claude": "anthropic/claude-3.5-sonnet:beta"
+                "Kimi": "moonshotai/kimi-dev-72b:free",
+                "Claude": "anthropic/claude-3.5-sonnet:beta",
+                "DeepSeek-R1": "deepseek/deepseek-r1-0528:free"
             }
             
             completion = client.chat.completions.create(
@@ -499,7 +486,7 @@ def create_ui() -> gr.Blocks:
                 with gr.Group(visible=True) as params_group:
                     gr.Markdown("### Exercise Parameters")
                     ai_model = gr.Radio(
-                        ["Mistral", "DeepSeek", "Gemma", "Claude"],
+                        ["Mistral", "Gemma", "Kimi", "Claude", "DeepSeek-R1"],
                         value="Mistral",
                         label="AI Model"
                     )