o d6i@sdZddlZddlmZddlmZmZddlmZmZddl m Z m Z m Z m Z mZmZeeZdeded ed efd d Zdeded ed efddZdededed efddZdeded efddZdS)z Utility functions for Karishye AI. This module contains: - Puja state management functions (get/save user puja state) - Phone number validation and saving - Puja booking utilities (cancellation, completion response) - Knowledge base helpers N)datetime)DictAny) general_agent ChatContext)create_puja_sessionget_puja_sessionupdate_puja_sessionpuja_session_to_dictsave_user_phone_numbervalidate_indian_phone_numberuser_id puja_type session_idreturncCs&t|||}|r t|S|diddS)z Get puja session state from database. Returns dict for backward compatibility with existing code. Session ID is required - no default fallback. rF)rcurrent_questioncollected_info completed)rr r rr puja_sessionr3/var/www/html/karishye-ai-python/app/utils/utils.pyget_user_puja_states rstatecszhdfdd|D}t|||}|r!t||||ddS|d|}t||||d}|r;t||||ddSdS)z Save puja session state to database. Accepts dict for backward compatibility with existing code. Session ID is required - no default fallback. >r kb_checkedawaiting_phonerphone_collectedrinvalid_attemptscsi|] \}}|vr||qSrr).0kv VALID_FIELDSrr @sz(save_user_puja_state..r rrupdates puja_name)r rrr&N)itemsrr getr)r rrrfiltered_stateexisting_sessionr& new_sessionrr!rsave_user_puja_state4s2   r, phone_numberc CsRz t||||WdSty(}ztdt|jWYd}~dSd}~ww)z@ Save user's phone number to database (NOT sent to AI). TzB] Failed to save phone: NF)save_phone_to_db Exceptionloggererrortype__name__)r r-rrerrrr gsr cCsJtd|d|d|t|||}|r#t|||dddddSdS) zNCancel the ongoing puja collection by marking it as cancelled in the database.z$Cancelling puja collection for user z, puja z , session TF)rrr$N)r0inforr rrrrcancel_puja_collectionws  r6)__doc__loggingrtypingrrapp.agents.agentsrrapp.utils.puja_session_utilsrrr r r r.r getLoggerr3r0strdictrr,r6rrrrs   3