Update model_api.py

model_api.py

@@ -1,39 +1,25 @@
 from huggingface_hub import InferenceClient
 import os
-import json
-import re
 
 def query_model(prompt):
     try:
-        import streamlit as st
-
-        HF_TOKEN = st.secrets["HF_TOKEN"]
+        HF_TOKEN = os.getenv("HF_TOKEN")
 
         client = InferenceClient(
-            provider="auto",
-            api_key=HF_TOKEN
+            model="meta-llama/Llama-3.2-3B-Instruct",
+            token=HF_TOKEN
         )
 
-        response = client.chat.completions.create(
-            model="Qwen/Qwen2.5-7B-Instruct",
+        response = client.chat_completion(
             messages=[
-                {"role": "system", "content": "You are a professional fitness trainer that returns ONLY JSON workout plans."},
+                {"role": "system", "content": "You are a certified professional fitness trainer."},
                 {"role": "user", "content": prompt}
             ],
-            max_tokens=1200,
-            temperature=0.6
+            max_tokens=2000,
+            temperature=0.7
         )
 
-        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}
+        return response.choices[0].message.content
 
     except Exception as e:
-        return {"error": str(e)}
+        return f"Error: {str(e)}"