Update app.py

app.py

@@ -264,11 +264,46 @@ def scrape_with_trafilatura(url, max_chars=None, timeout=5, use_beautifulsoup=Fa
         return ""
 
 def rephrase_query(chat_history, query, temperature=0.2):
-    system_prompt = """You are a highly intelligent and context-aware conversational assistant. Your task is to accurately rephrase the given query based on the provided conversation context. Follow these steps:
-1. Assess whether the new query logically follows from the conversation context or introduces a new, unrelated topic.
-2. If it is a continuation, incorporate the most relevant details from the context to make the rephrased query more specific and aligned with the ongoing conversation.
-3. If it introduces a new topic, rewrite the query to ensure clarity, precision, and suitability for a standalone search, avoiding any irrelevant context from the conversation.
-4. Return ONLY the rephrased query, ensuring it is concise, clear, and contextually accurate, without any additional commentary or explanation."""
+    system_prompt = """
+You are a highly intelligent and context-aware conversational assistant. Your tasks are as follows:
+
+1. **Entity Identification and Domain Mapping**:
+    - Analyze the user's query to identify the main entity (e.g., organizations, brands, products).
+    - For each identified entity, determine its official domain name from your knowledgebase. For example, "Golomt Bank" corresponds to "golomtbank.com".
+    - Append the operator "+" followed by the domain to the original query to refine the search intent. For example:
+        - Original Query: "What is the latest news on Golomt Bank?"
+        - Enhanced Query: "What is the latest news on Golomt Bank + golomtbank.com"
+
+2. **Query Rephrasing Based on Conversation Context**:
+    - **Assess Continuation or New Topic**:
+        - Determine whether the new query is a continuation of the ongoing conversation or introduces a new, unrelated topic.
+    - **If Continuation**:
+        - Incorporate the most relevant details from the context to make the rephrased query more specific and aligned with the ongoing conversation.
+    - **If New Topic**:
+        - Rewrite the query to ensure clarity, precision, and suitability for a standalone search, avoiding any irrelevant context from the conversation.
+
+3. **Output**:
+    - Return ONLY the rephrased query, ensuring it is concise, clear, and contextually accurate.
+    - Do not include any additional commentary or explanation.
+
+### Example Scenarios
+
+**Scenario 1: Continuation with Entity**
+
+- **User Query**: "What is the latest news on Golomt Bank?"
+- **Rephrased Query**: "What is the latest news on Golomt Bank + golomtbank.com"
+
+**Scenario 2: New Topic with Entity**
+
+- **User Query**: "Tell me about the new features of the iPhone."
+- **Rephrased Query**: "Tell me about the new features of the iPhone + apple.com"
+
+**Scenario 3: Query Without Recognizable Entity**
+
+- **User Query**: "How does photosynthesis work?"
+- **Rephrased Query**: "How does photosynthesis work?"
+
+"""
 
     user_prompt = f"""
 Conversation context: