Update app.py

app.py

@@ -628,41 +628,24 @@ def rephrase_for_search(query, model):
     rephrase_prompt = PromptTemplate(
         input_variables=["query"],
         template="""
-        Rephrase the following conversational query into a concise, search-engine-friendly format.
-        Maintain key terms and add relevant context if necessary.
-        Provide ONLY the rephrased query without any explanation or additional text.
-
+        Your task is to rephrase the given conversational query into a concise, search-engine-friendly format.
+        Remove any conversational elements and focus on the core information need.
+        
         Conversational query: {query}
-
-        Rephrased query:"""
+        
+        Rephrased query: """
     )
     
     chain = LLMChain(llm=model, prompt=rephrase_prompt)
     response = chain.run(query=query).strip()
     
-    # Extract only the rephrased query using regex
-    match = re.search(r'^(.*?)(?:\n|$)', response)
-    if match:
-        rephrased_query = match.group(1).strip()
-    else:
-        rephrased_query = response.strip()
-    
-    # Remove any "Rephrased query:" prefix if present
-    rephrased_query = re.sub(r'^Rephrased query:\s*', '', rephrased_query, flags=re.IGNORECASE)
-    
-    # Check if the rephrased query is actually a rephrasing and not the original prompt or instructions
-    if (rephrased_query.lower().startswith(("rephrase", "your task")) or 
-        len(rephrased_query.split()) > len(query.split()) * 2):
-        # If it's not a proper rephrasing, use a more sophisticated keyword extraction
-        keywords = query.lower()
-        # Remove common stop words and question words
-        stop_words = {'how', 'did', 'the', 'in', 'a', 'an', 'and', 'or', 'but', 'is', 'are', 'was', 'were', 'perform'}
-        keywords = ' '.join(word for word in keywords.split() if word not in stop_words)
-        # Ensure important terms are included
-        if 'kkr' not in keywords:
-            keywords = 'kkr ' + keywords
-        if 'q1 2024' not in keywords:
-            keywords += ' q1 2024'
+    # Extract only the rephrased query, ignoring any explanations
+    rephrased_query = response.split('\n')[0].strip()
+    
+    # Check if the rephrased query is actually a rephrasing and not the original prompt
+    if rephrased_query.lower().startswith("rephrase") or len(rephrased_query) > len(query) * 2:
+        # If it's not a proper rephrasing, use a simple keyword extraction
+        keywords = ' '.join(query.lower().split())
         return keywords
     
     return rephrased_query