Update app.py

app.py

@@ -25,14 +25,14 @@ def extract_suggestion_value(text):
     try:
         parsed = json.loads(text)
         if "suggestion" in parsed:
-            return {"suggestion": parsed["suggestion"]}
+            return parsed["suggestion"]
     except json.JSONDecodeError:
         pass  # Fall back to regex if not valid JSON
 
     # Fallback: use regex to extract suggestion value
     match = re.search(r'"suggestion"\s*:\s*"([^"]+)"', text)
     if match:
-        return {"suggestion": match.group(1)}
+        return match.group(1)
 
     return None
 
@@ -59,15 +59,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.")
 
-        print(content)
-        
         content = extract_suggestion_value(content)
 
-        print("----")
-
-        print(content)
-
-        return content
+        return content.strip()
         
     except requests.exceptions.RequestException as e:
         return f"API Request Error: {e}"