Update prompt

main.py

@@ -158,8 +158,9 @@ class HealthcareChatbot:
                 template="""
             You are a routing system. Your job is simple:
             1. Understand what the user wants
-            2. Check if any endpoint can do what they want
-            3. If yes = API_ACTION, if no = CONVERSATION
+            2. Handle any dates/times in their request
+            3. Check if any endpoint can do what they want
+            4. If yes = API_ACTION, if no = CONVERSATION
 
             ## Available API Endpoints Documentation
             {endpoints_documentation}
@@ -167,6 +168,9 @@ class HealthcareChatbot:
             ## User Query to Analyze
             Query: "{user_query}"
             Language: {detected_language}
+            Current Context:
+            - DateTime: {current_datetime}
+            - Timezone: {timezone}
 
             ## Step-by-Step Analysis
 
@@ -175,7 +179,22 @@ class HealthcareChatbot:
             - Identify the exact action or information the user is requesting
             - Focus on understanding their underlying need, not just the words
 
-            **STEP 2: Find matching endpoint**
+            **STEP 2: Handle Date/Time Processing**
+            - Identify any temporal expressions in the user query
+            - Convert relative dates/times using the current context:
+              * "اليوم" (today) = current date
+              * "غدا" (tomorrow) = current date + 1 day  
+              * "أمس" (yesterday) = current date - 1 day
+              * "الأسبوع القادم" (next week) = current date + 7 days
+              * "بعد ساعتين" (in 2 hours) = current time + 2 hours
+              * "صباحًا" (morning/AM), "مساءً" (evening/PM)
+              * "الشهر القادم" (next month) = current date + 1 month
+              * "الأسبوع الماضي" (last week) = current date - 7 days
+            - Handle different date formats and languages
+            - Account for timezone differences
+            - Convert to ISO 8601 format: YYYY-MM-DDTHH:MM:SS
+
+            **STEP 3: Find matching endpoint**
             - Read each endpoint description in the documentation
             - Check if any endpoint's purpose can fulfill what the user wants
             - Match based on functionality, not keywords
@@ -184,11 +203,16 @@ class HealthcareChatbot:
             - Found matching endpoint = "API_ACTION"
             - No matching endpoint = "CONVERSATION"
 
+            **STEP 4: Parameter Extraction (only if API_ACTION)**
+            - Extract parameter values from user query
+            - Convert dates/times to ISO 8601 format (YYYY-MM-DDTHH:MM:SS) using the context above
+            - List any missing required parameters
+
             ## Output Format
             {{
                 "intent": "CONVERSATION|API_ACTION",
                 "confidence": 0.8,
-                "reasoning": "User wants: [what user actually needs]. Found endpoint: [endpoint path and why it matches] OR No endpoint matches this need",
+                "reasoning": "User wants: [what user actually needs]. Date/time processing: [any date conversions made]. Found endpoint: [endpoint path and why it matches] OR No endpoint matches this need",
                 "endpoint": "/exact/endpoint/path",
                 "method": "GET|POST|PUT|DELETE",
                 "params": {{}},
@@ -198,8 +222,54 @@ class HealthcareChatbot:
             Now analyze the user query step by step and give me the JSON response.
             """,
                 input_variables=["user_query", "detected_language", "extracted_keywords", 
-                                "sentiment_analysis", "endpoints_documentation"]
+                                "sentiment_analysis", "endpoints_documentation", "current_datetime", "timezone"]
             )
+        # self.router_prompt_template = PromptTemplate(
+        #         template="""
+        #     You are a routing system. Your job is simple:
+        #     1. Understand what the user wants
+        #     2. Check if any endpoint can do what they want
+        #     3. If yes = API_ACTION, if no = CONVERSATION
+
+        #     ## Available API Endpoints Documentation
+        #     {endpoints_documentation}
+
+        #     ## User Query to Analyze
+        #     Query: "{user_query}"
+        #     Language: {detected_language}
+
+        #     ## Step-by-Step Analysis
+
+        #     **STEP 1: What does the user want?**
+        #     - If query is in Arabic, translate it to English first
+        #     - Identify the exact action or information the user is requesting
+        #     - Focus on understanding their underlying need, not just the words
+
+        #     **STEP 2: Find matching endpoint**
+        #     - Read each endpoint description in the documentation
+        #     - Check if any endpoint's purpose can fulfill what the user wants
+        #     - Match based on functionality, not keywords
+
+        #     **STEP 3: Decision**
+        #     - Found matching endpoint = "API_ACTION"
+        #     - No matching endpoint = "CONVERSATION"
+
+        #     ## Output Format
+        #     {{
+        #         "intent": "CONVERSATION|API_ACTION",
+        #         "confidence": 0.8,
+        #         "reasoning": "User wants: [what user actually needs]. Found endpoint: [endpoint path and why it matches] OR No endpoint matches this need",
+        #         "endpoint": "/exact/endpoint/path",
+        #         "method": "GET|POST|PUT|DELETE",
+        #         "params": {{}},
+        #         "missing_required": []
+        #     }}
+
+        #     Now analyze the user query step by step and give me the JSON response.
+        #     """,
+        #         input_variables=["user_query", "detected_language", "extracted_keywords", 
+        #                         "sentiment_analysis", "endpoints_documentation"]
+        #     )
         # CONVERSATION CHAIN - Handles conversational responses
         self.conversation_template = PromptTemplate(
             template="""
@@ -666,7 +736,9 @@ class HealthcareChatbot:
                 "extracted_keywords": json.dumps(keywords),
                 "sentiment_analysis": json.dumps(sentiment_result),
                 "conversation_history": self.get_conversation_context(),
-                "endpoints_documentation": json.dumps(self.endpoints_documentation, indent=2)
+                "endpoints_documentation": json.dumps(self.endpoints_documentation, indent=2),
+                "current_datetime": datetime.now().strftime('%Y-%m-%dT%H:%M:%S'),
+                "timezone": "UTC",
             })
             
             # Parse router response