Update model_api.py

model_api.py

@@ -1,21 +1,37 @@
 from huggingface_hub import InferenceClient
 import os
+import json
+import re
 
 def query_model(prompt):
     try:
         HF_TOKEN = os.getenv("HF_TOKEN")
-        # You can also set provider at the client level
-        client = InferenceClient(api_key=HF_TOKEN, provider="auto")
-        
+
+        client = InferenceClient(
+            provider="auto",
+            api_key=HF_TOKEN
+        )
+
         response = client.chat.completions.create(
             model="Qwen/Qwen2.5-7B-Instruct",
             messages=[
-                {"role": "system", "content": "You are a professional fitness trainer."},
+                {"role": "system", "content": "You are a professional fitness trainer that returns ONLY JSON workout plans."},
                 {"role": "user", "content": prompt}
             ],
-            max_tokens=1500,
-            temperature=0.7
+            max_tokens=1200,
+            temperature=0.6
         )
-        return response.choices[0].message.content
+
+        result = response.choices[0].message.content
+
+        # Remove markdown JSON blocks if model returns ```json ```
+        result = re.sub(r"```json|```", "", result).strip()
+
+        try:
+            result_json = json.loads(result)
+            return result_json
+        except json.JSONDecodeError:
+            return {"error": "Model returned invalid JSON", "raw_output": result}
+
     except Exception as e:
-        return f"Error: {str(e)}"
+        return {"error": str(e)}