Update app.py

app.py

@@ -42,6 +42,27 @@ def get_damaged_object_estimates(damage_description):
 
     return response
 
+def postprocess_message_text(api_response):
+    """
+    Extracts and formats the message content from the API response JSON.
+    
+    Args:
+        api_response (str): JSON string from the API response.
+        
+    Returns:
+        str: Formatted message content.
+    """
+    # Parse the JSON response
+    response = json.loads(api_response)
+    
+    # Extract the content from the choices
+    message_content = response["choices"][0]["message"]["content"]
+    
+    # Clean the message content (remove unnecessary escape characters)
+    formatted_content = message_content.replace("\\n", "\n")
+    
+    return formatted_content
+
 
 # Streamlit interface
 st.title("Disaster Damage Cost Estimator")
@@ -54,7 +75,7 @@ if st.button("Get Estimated Costs"):
     if damage_description:
         with st.spinner("Fetching cost estimates..."):
             # Call the function to get estimates
-            estimated_costs = get_damaged_object_estimates(damage_description)
+            estimated_costs = postprocess_message_text(get_damaged_object_estimates(damage_description))
             
             if estimated_costs:
                 st.success("Here are the estimated costs:")