Update app.py

app.py

@@ -111,10 +111,8 @@ class DocumentRetrievalAndGeneration:
         if match1:
             solution_text = match1.group(1).strip()
             print(solution_text)
-            match12 = re.search(r'Solution:(.*?)', solution_text, re.DOTALL)
-            if match12:
-                solution_text=match12.group(1).strip()
-                print(solution_text)
+            if "Solution:" in solution_text:
+            solution_text = solution_text.split("Solution:", 1)[1].strip()
         elif match2:
             solution_text = match2.group(1).strip()
             print(solution_text)