Update app.py

app.py

@@ -8,6 +8,15 @@ import re
 API_URL = "https://deployment.datasaur.ai/api/deployment/8/2723/chat/completions"
 API_TOKEN = os.environ["DATASAUR_API_KEY"]
 
+def extract_suggestion_with_regex(xml_string):
+    """Extract content of <suggestion> tag using regex."""
+    pattern = r"<suggestion>(.*?)</suggestion>"
+    match = re.search(pattern, xml_string, re.DOTALL)
+    if match:
+        return match.group(1).strip()
+    else:
+        raise ValueError("No <suggestion> tag found")
+        
 def magic_function(input_text):
     """
     Sends text to the Datasaur deployment API and returns the processed text.
@@ -30,12 +39,9 @@ def magic_function(input_text):
         # This may need adjustment if the API has a different format.
         content = response_json.get("choices", [{}])[0].get("message", {}).get("content", "Error: Could not parse response.")
         
-        # Remove unwanted prefixes and characters from the response.
-        # This regex removes "Suggestion: " or "**Suggestion**: " from the beginning of the string, case-insensitively.
-        content = re.sub(r'^\s*(\*\*Suggestion\*\*:\s*|Suggestion:\s*)', '', content, flags=re.IGNORECASE)
-        content = content.replace("`", "")
-        
-        return content.strip()
+        content = extract_suggestion_with_regex(content)
+
+        return content.strip()   
         
     except requests.exceptions.RequestException as e:
         return f"API Request Error: {e}"