made changes in inference.py file

app/inference_api.py

@@ -54,26 +54,47 @@ RECOMMENDATIONS:
                     "processing_time": processing_time
                 }
 
-            # Trim to start from SAFETY STATUS (case-insensitive)
+            # Clean up the response - remove unwanted instructional text
+            cleaned_response = raw_response
+            
+            # Remove the specific problematic text and similar variations
+            unwanted_texts = [
+                "If there is no need for recommendations, don't provide them.",
+                "If there is no need for recommendations, don't provide them",
+                "If there is no need for recommendations, do not provide them.",
+                "If there is no need for recommendations, do not provide them",
+                "If no recommendations are needed, don't provide them.",
+                "If no recommendations are needed, don't provide them",
+                "Don't provide recommendations if not needed.",
+                "Don't provide recommendations if not needed"
+            ]
+            
+            for unwanted_text in unwanted_texts:
+                cleaned_response = cleaned_response.replace(unwanted_text, "").strip()
+            
+            # Remove any text that appears before the actual analysis
+            # Look for patterns like "OVERVIEW", "ANALYSIS", etc. followed by the unwanted text
             import re
-            match = re.search(r"(?i)safety\s*status\s*:", raw_response)
+            
+            # Remove any instructional text that might appear at the beginning
+            cleaned_response = re.sub(r'^.*?(?=SAFETY\s*STATUS\s*:|SAFETY\s*ASSESSMENT\s*:)', '', cleaned_response, flags=re.IGNORECASE | re.DOTALL).strip()
+            
+            # Try to find SAFETY STATUS first
+            match = re.search(r"(?i)safety\s*status\s*:", cleaned_response)
             if match:
-                cleaned_response = raw_response[match.start():].strip()
-                # Verify the output has the required structure
-                if not re.search(r"(?i)safety\s*assessment\s*:", cleaned_response) and not re.search(r"(?i)recommendations\s*:", cleaned_response):
-                    # Missing required sections, return default message
-                    return {
-                        "response": "Model failed to generate proper analysis. Please try again.",
-                        "status": "success",
-                        "model_used": "huggingface-qwen-fallback",
-                        "prompt": prompt,
-                        "processing_time": processing_time
-                    }
+                cleaned_response = cleaned_response[match.start():].strip()
             else:
-                # No SAFETY STATUS found, return default message
+                # If no SAFETY STATUS, try to find SAFETY ASSESSMENT
+                match = re.search(r"(?i)safety\s*assessment\s*:", cleaned_response)
+                if match:
+                    cleaned_response = "SAFETY STATUS: SAFE\n\n" + cleaned_response[match.start():].strip()
+            
+            # Verify the output has the required structure
+            if not re.search(r"(?i)safety\s*assessment\s*:", cleaned_response):
+                # Missing required sections, return default message
                 return {
                     "response": "Model failed to generate proper analysis. Please try again.",
-                    "status": "success",
+                    "status": "success", 
                     "model_used": "huggingface-qwen-fallback",
                     "prompt": prompt,
                     "processing_time": processing_time