Update app.py

app.py

@@ -8,6 +8,7 @@ AND time signature, guaranteeing exact durations in MIDI and in the UI!
 Major updates:
 - 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
 """
@@ -28,7 +29,7 @@ def install(packages: List[str]):
             subprocess.check_call([sys.executable, "-m", "pip", "install", package])
 
 install([
-    "mido", "midi2audio", "pydub", "gradio",
+    "mido", "midi2audio", "pydub", "gradio", "openai",
     "requests", "numpy", "matplotlib", "librosa", "scipy",
 ])
 
@@ -54,12 +55,14 @@ import os
 import subprocess as sp
 import base64
 import shutil
+from openai import OpenAI  # For DeepSeek API
 
 # -----------------------------------------------------------------------------
 # 3. Configuration & constants
 # -----------------------------------------------------------------------------
 MISTRAL_API_URL = "https://api.mistral.ai/v1/chat/completions"
-MISTRAL_API_KEY = "yQdfM8MLbX9uhInQ7id4iUTwN4h4pDLX"  # ← Replace with your key!
+MISTRAL_API_KEY = "yQdfM8MLbX9uhInQ7id4iUTwN4h4pDLX"  # Replace with your key
+DEEPSEEK_API_KEY = "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"
 
 SOUNDFONT_URLS = {
     "Trumpet": "https://github.com/FluidSynth/fluidsynth/raw/master/sf2/Trumpet.sf2",
@@ -258,14 +261,10 @@ def get_technique_based_on_level(level: str) -> str:
     return random.choice(techniques.get(level, ["with slurs"]))
 
 # -----------------------------------------------------------------------------
-# 9. Mistral API: query, fallback on errors (using integer durations)
+# 9. LLM Query Function (supports Mistral and DeepSeek)
 # -----------------------------------------------------------------------------
-def query_mistral(prompt: str, instrument: str, level: str, key: str,
-                  time_sig: str, measures: int) -> str:
-    headers = {
-        "Authorization": f"Bearer {MISTRAL_API_KEY}",
-        "Content-Type": "application/json",
-    }
+def query_llm(model_name: str, prompt: str, instrument: str, level: str, key: str,
+              time_sig: str, measures: int) -> str:
     numerator, denominator = map(int, time_sig.split('/'))
     units_per_measure = numerator * (16 // denominator)
     required_total = measures * units_per_measure
@@ -296,26 +295,62 @@ def query_mistral(prompt: str, instrument: str, level: str, key: str,
             "Sum must be exactly as specified. ONLY output the JSON array. No prose."
         )
 
-    payload = {
-        "model": "mistral-medium",
-        "messages": [
-            {"role": "system", "content": system_prompt},
-            {"role": "user", "content": user_prompt},
-        ],
-        "temperature": 0.7 if level == "Advanced" else 0.5,
-        "max_tokens": 1000,
-        "top_p": 0.95,
-        "frequency_penalty": 0.2,
-        "presence_penalty": 0.2,
-    }
-
-    try:
-        response = requests.post(MISTRAL_API_URL, headers=headers, json=payload)
-        response.raise_for_status()
-        content = response.json()["choices"][0]["message"]["content"]
-        return content.replace("```json","").replace("```","").strip()
-    except Exception as e:
-        print(f"Error querying Mistral API: {e}")
+    if model_name == "Mistral":
+        headers = {
+            "Authorization": f"Bearer {MISTRAL_API_KEY}",
+            "Content-Type": "application/json",
+        }
+        payload = {
+            "model": "mistral-medium",
+            "messages": [
+                {"role": "system", "content": system_prompt},
+                {"role": "user", "content": user_prompt},
+            ],
+            "temperature": 0.7 if level == "Advanced" else 0.5,
+            "max_tokens": 1000,
+            "top_p": 0.95,
+            "frequency_penalty": 0.2,
+            "presence_penalty": 0.2,
+        }
+        try:
+            response = requests.post(MISTRAL_API_URL, headers=headers, json=payload)
+            response.raise_for_status()
+            content = response.json()["choices"][0]["message"]["content"]
+            return content.replace("```json","").replace("```","").strip()
+        except Exception as e:
+            print(f"Error querying Mistral API: {e}")
+            return get_fallback_exercise(instrument, level, key, time_sig, measures)
+    
+    elif model_name == "DeepSeek":
+        try:
+            client = OpenAI(
+                base_url="https://openrouter.ai/api/v1",
+                api_key=DEEPSEEK_API_KEY,
+            )
+            
+            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",
+                messages=[
+                    {"role": "system", "content": system_prompt},
+                    {"role": "user", "content": user_prompt},
+                ],
+                temperature=0.7 if level == "Advanced" else 0.5,
+                max_tokens=1000,
+                top_p=0.95,
+                frequency_penalty=0.2,
+                presence_penalty=0.2,
+            )
+            content = completion.choices[0].message.content
+            return content.replace("```json","").replace("```","").strip()
+        except Exception as e:
+            print(f"Error querying DeepSeek API: {e}")
+            return get_fallback_exercise(instrument, level, key, time_sig, measures)
+    
+    else:
         return get_fallback_exercise(instrument, level, key, time_sig, measures)
 
 # -----------------------------------------------------------------------------
@@ -336,10 +371,10 @@ def safe_parse_json(text: str) -> Optional[list]:
 # 11. Main orchestration: talk to API, *scale durations*, build MIDI, UI values
 # -----------------------------------------------------------------------------
 def generate_exercise(instrument: str, level: str, key: str, tempo: int, time_signature: str,
-                      measures: int, custom_prompt: str, mode: str) -> Tuple[str, Optional[str], str, MidiFile, str, str, int]:
+                      measures: int, custom_prompt: str, mode: str, ai_model: str) -> Tuple[str, Optional[str], str, MidiFile, str, str, int]:
     try:
         prompt_to_use = custom_prompt if mode == "Exercise Prompt" else ""
-        output = query_mistral(prompt_to_use, instrument, level, key, time_signature, measures)
+        output = query_llm(ai_model, prompt_to_use, instrument, level, key, time_signature, measures)
         parsed = safe_parse_json(output)
         if not parsed:
             return "Invalid JSON format", None, str(tempo), None, "0", time_signature, 0
@@ -366,7 +401,7 @@ def generate_exercise(instrument: str, level: str, key: str, tempo: int, time_si
 # -----------------------------------------------------------------------------
 # 12. Simple AI chat assistant (optional, shares LLM)
 # -----------------------------------------------------------------------------
-def handle_chat(message: str, history: List, instrument: str, level: str):
+def handle_chat(message: str, history: List, instrument: str, level: str, ai_model: str):
     if not message.strip():
         return "", history
     messages = [{"role": "system", "content": f"You are a {instrument} teacher for {level} students."}]
@@ -374,16 +409,46 @@ def handle_chat(message: str, history: List, instrument: str, level: str):
         messages.append({"role": "user", "content": user_msg})
         messages.append({"role": "assistant", "content": assistant_msg})
     messages.append({"role": "user", "content": message})
-    headers = {"Authorization": f"Bearer {MISTRAL_API_KEY}", "Content-Type": "application/json"}
-    payload = {"model": "mistral-medium", "messages": messages, "temperature": 0.7, "max_tokens": 500}
-    try:
-        response = requests.post(MISTRAL_API_URL, headers=headers, json=payload)
-        response.raise_for_status()
-        content = response.json()["choices"][0]["message"]["content"]
-        history.append((message, content))
-        return "", history
-    except Exception as e:
-        history.append((message, f"Error: {str(e)}"))
+    
+    if ai_model == "Mistral":
+        headers = {"Authorization": f"Bearer {MISTRAL_API_KEY}", "Content-Type": "application/json"}
+        payload = {"model": "mistral-medium", "messages": messages, "temperature": 0.7, "max_tokens": 500}
+        try:
+            response = requests.post(MISTRAL_API_URL, headers=headers, json=payload)
+            response.raise_for_status()
+            content = response.json()["choices"][0]["message"]["content"]
+            history.append((message, content))
+            return "", history
+        except Exception as e:
+            history.append((message, f"Error: {str(e)}"))
+            return "", history
+    
+    elif ai_model == "DeepSeek":
+        try:
+            client = OpenAI(
+                base_url="https://openrouter.ai/api/v1",
+                api_key=DEEPSEEK_API_KEY,
+            )
+            
+            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",
+                messages=messages,
+                temperature=0.7,
+                max_tokens=500,
+            )
+            content = completion.choices[0].message.content
+            history.append((message, content))
+            return "", history
+        except Exception as e:
+            history.append((message, f"Error: {str(e)}"))
+            return "", history
+    
+    else:
+        history.append((message, "Error: Invalid AI model selected"))
         return "", history
 
 # -----------------------------------------------------------------------------
@@ -400,6 +465,11 @@ def create_ui() -> gr.Blocks:
             with gr.Column(scale=1):
                 with gr.Group(visible=True) as params_group:
                     gr.Markdown("### Exercise Parameters")
+                    ai_model = gr.Radio(
+                        ["Mistral", "DeepSeek"],
+                        value="Mistral",
+                        label="AI Model"
+                    )
                     instrument = gr.Dropdown([
                         "Trumpet", "Piano", "Violin", "Clarinet", "Flute",
                     ], value="Trumpet", label="Instrument")
@@ -425,7 +495,7 @@ def create_ui() -> gr.Blocks:
                         duration_display = gr.Textbox(label="Audio Duration", interactive=False)
                     with gr.TabItem("Exercise Data"):
                         json_output = gr.Code(label="JSON Representation", language="json")
-                        # NEW: Duration sum display
+                        # Duration sum display
                         duration_sum = gr.Number(
                             label="Total Duration Units (16th notes)",
                             interactive=False,
@@ -447,16 +517,16 @@ def create_ui() -> gr.Blocks:
             inputs=[mode], outputs=[params_group, prompt_group]
         )
         def generate_caller(mode_val, instrument_val, level_val, key_val,
-                    time_sig_val, measures_val, prompt_val, measures_prompt_val):
+                    time_sig_val, measures_val, prompt_val, measures_prompt_val, ai_model_val):
             real_measures = measures_prompt_val if mode_val == "Exercise Prompt" else measures_val
             fixed_tempo = 60
             return generate_exercise(
                 instrument_val, level_val, key_val, fixed_tempo, time_sig_val,
-                real_measures, prompt_val, mode_val
+                real_measures, prompt_val, mode_val, ai_model_val
             )
         generate_btn.click(
             fn=generate_caller,
-            inputs=[mode, instrument, level, key, time_signature,measures, custom_prompt, measures_prompt],
+            inputs=[mode, instrument, level, key, time_signature, measures, custom_prompt, measures_prompt, ai_model],
             outputs=[json_output, audio_output, bpm_display, current_midi, duration_display, time_sig_display, duration_sum]
         )
         def save_midi(json_data, instr, time_sig):
@@ -479,7 +549,7 @@ def create_ui() -> gr.Blocks:
         )
         send_chat_btn.click(
             fn=handle_chat,
-            inputs=[chat_message, chat_history, instrument, level],
+            inputs=[chat_message, chat_history, instrument, level, ai_model],
             outputs=[chat_message, chat_history],
         )
     return demo
@@ -489,4 +559,4 @@ def create_ui() -> gr.Blocks:
 # -----------------------------------------------------------------------------
 if __name__ == "__main__":
     demo = create_ui()
-    demo.launch()
+    demo.launch()