Spaces:
Running
Running
File size: 2,749 Bytes
75e2b6c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
import json
import logging
from app.services.chathistory import ChatSession
from app.services.llm_model import Model
from app.services.environmental_condition import EnvironmentalData
from app.services.prompts import SKIN_CARE_SCHEDULER, DEFAULT_SCHEDULE
class SkinCareScheduler:
def __init__(self, token, session_id):
self.token = token
self.session_id = session_id
self.chat_session = ChatSession(token, session_id)
self.user_city = self.chat_session.get_city() or ''
self.environment_data = EnvironmentalData(self.user_city)
def get_historical_data(self):
"""Retrieve the last 7 days of schedules."""
schedules = self.chat_session.get_last_seven_days_schedules()
return [schedule["schedule_data"] for schedule in schedules]
def createTable(self):
"""Generate and return a daily skincare schedule."""
try:
# Check for an existing valid schedule
existing_schedule = self.chat_session.get_today_schedule()
if existing_schedule and isinstance(existing_schedule.get("schedule_data"), list):
return json.dumps(existing_schedule["schedule_data"], indent=2)
# Gather input data
historical_data = self.get_historical_data()
personalized_condition = self.chat_session.get_personalized_recommendation() or "No specific skin conditions provided"
environmental_data = self.environment_data.get_environmental_data()
# Format the prompt
formatted_prompt = SKIN_CARE_SCHEDULER.format(
personalized_condition=personalized_condition,
environmental_values=json.dumps(environmental_data, indent=2),
historical_data=json.dumps(historical_data, indent=2)
)
# Generate schedule with the model
model = Model()
result = model.skinScheduler(formatted_prompt)
# Handle errors by falling back to default schedule
if isinstance(result, dict) and "error" in result:
logging.error(f"Model error: {result['error']}")
result = DEFAULT_SCHEDULE
# Validate basic structure (optional, but ensures 5 entries)
if not isinstance(result, list) or len(result) != 5:
logging.warning("Generated schedule invalid; using default.")
result = DEFAULT_SCHEDULE
# Save and return the schedule
self.chat_session.save_schedule(result)
return json.dumps(result, indent=2)
except Exception as e:
logging.error(f"Schedule generation failed: {str(e)}")
return json.dumps(DEFAULT_SCHEDULE, indent=2) |