Update app.py

app.py

@@ -26,31 +26,31 @@ class Agent1:
         self.model = model
 
     def rephrase_and_split(self, user_input: str) -> List[str]:
-    rephrase_prompt = PromptTemplate(
-        input_variables=["query"],
-        template="""
-        Rephrase the given query into one or more concise, search-engine-friendly formats.
-        If the query contains multiple distinct questions, split them.
-        Provide ONLY the rephrased queries without any additional text or explanations, one per line.
-        
-        Query: {query}
-        
-        Rephrased queries:"""
-    )
-    
-    chain = LLMChain(llm=self.model, prompt=rephrase_prompt)
-    response = chain.run(query=user_input).strip()
-    
-    # Split the response at "Rephrased queries:" and take the second part
-    split_response = response.split("Rephrased queries:", 1)
-    if len(split_response) > 1:
-        response = split_response[1].strip()
-    
-    # Remove any lines that contain instructions or explanations
-    rephrased_queries = [q.strip() for q in response.split('\n') if q.strip() and not q.startswith("Rephrase") and "query" not in q.lower()]
-    
-    # If no valid rephrased queries, return the original input
-    return rephrased_queries if rephrased_queries else [user_input]
+        rephrase_prompt = PromptTemplate(
+            input_variables=["query"],
+            template="""
+            Rephrase the given query into one or more concise, search-engine-friendly formats.
+            If the query contains multiple distinct questions, split them.
+            Provide ONLY the rephrased queries without any additional text or explanations, one per line.
+
+            Query: {query}
+
+            Rephrased queries:"""
+        )
+
+        chain = LLMChain(llm=self.model, prompt=rephrase_prompt)
+        response = chain.run(query=user_input).strip()
+
+        # Split the response at "Rephrased queries:" and take the second part
+        split_response = response.split("Rephrased queries:", 1)
+        if len(split_response) > 1:
+            response = split_response[1].strip()
+
+        # Remove any lines that contain instructions or explanations
+        rephrased_queries = [q.strip() for q in response.split('\n') if q.strip() and not q.startswith("Rephrase") and "query" not in q.lower()]
+
+        # If no valid rephrased queries, return the original input
+        return rephrased_queries if rephrased_queries else [user_input]
 
     def process(self, user_input: str) -> Dict[str, List[Dict[str, str]]]:
         queries = self.rephrase_and_split(user_input)