Update lifestyle_agent.py

lifestyle_agent.py

@@ -1,18 +1,11 @@
-def recommend_lifestyle(symptom: str, value: str):
-    if symptom.lower() == "high blood pressure":
-        return f"Recommended lifestyle for {symptom}: Reduce salt intake, avoid stress, exercise regularly, and monitor your BP."
-    elif symptom.lower() == "diabetes":
-        return f"Recommended lifestyle for {symptom}: Maintain a low-carb diet, monitor blood sugar, exercise, and avoid processed foods."
-    return "No recommendations available."
-
-def get_fitness_tips():
-    return "Try to engage in at least 30 minutes of physical activity daily. Regular exercise helps to lower blood pressure and improve overall health."
-
-def get_sleep_advice():
-    return "Ensure you get at least 7-8 hours of sleep each night. Avoid caffeine and screen time before bedtime."
-
-def get_diet_tips():
-    return "Eat a balanced diet with plenty of vegetables, fruits, and whole grains. Avoid high-sugar and high-fat foods."
-
-def get_general_suggestions():
-    return "It is important to maintain a balanced lifestyle: Eat healthy, stay active, and manage stress effectively."
+from transformers import AutoTokenizer, AutoModelForCausalLM
+import torch
+
+def get_lifestyle_tips(lifestyle_query):
+    model_name = "m42-health/med42-70b"
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+    prompt = f"Give healthy lifestyle tips based on the query: {lifestyle_query}."
+    inputs = tokenizer(prompt, return_tensors="pt")
+    outputs = model.generate(**inputs, max_new_tokens=150)
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)