Update app/utils/groq_client.py

app/utils/groq_client.py

@@ -94,12 +94,8 @@ class GroqClient:
 
         Example JSON structure (fill with relevant details based on input):
         {{ 
-            "caseSummary": {{ "caseID": "C-XXXX", "crimeType": "Type of Crime", "status": "Current Status", "briefOverview": "A brief overview of the case." }},
-            "incidentDetails": {{ "date": "YYYY-MM-DD", "time": "HH:MM", "location": "Location of incident", "narrative": "Detailed narrative of the incident." }},
-            "personsInvolved": {{ "suspects": [{{ "name": "Suspect Name", "role": "Suspect", "details": "Relevant details" }}], "victims": [{{ "name": "Victim Name", "role": "Victim", "details": "Relevant details" }}], "witnesses": [{{ "name": "Witness Name", "role": "Witness", "details": "Relevant details" }}] }},
-            "evidenceCollected": {{ "physical": ["Item 1", "Item 2"], "testimonial": ["Statement 1"], "digital": ["File A.jpg"] }},
-            "interrogationSummary": {{ "subjectName": "Suspect Name", "dateOfInterrogation": "YYYY-MM-DD", "keyStatements": ["Statement A", "Statement B"], "observations": "Behavioral observations during interrogation." }},
-            "findingsAndInferences": {{ "keyFindings": ["Finding 1", "Finding 2"], "inferences": ["Inference A", "Inference B"], "investigatorAnalysis": "Overall analysis by the investigator." }},
+            "caseSummary": {{ "observations": "Behavioral observations during interrogation." }},
+    
             "recommendations": {{ "chargesToBeFiled": ["Charge 1 (Specific to {selected_country_name} law)"], "furtherActions": ["Action A", "Action B (Consider {selected_country_name} procedures)"], "legalConsiderations_{selected_country_name}": "Specific legal points relevant to {selected_country_name}." }}
         }}
         """
@@ -122,7 +118,9 @@ class GroqClient:
                 top_p=1,
                 stop=None,
             )
-            json_output = chat_completion.choices[0].message.content
+            text_output = chat_completion.choices[0].message.content
+            print(text_output)
+            json_output = extract_json(text_output)
             # Validate if the output is valid JSON
             try:
                 json.loads(json_output)
@@ -145,6 +143,13 @@ class GroqClient:
                 "details": str(e)
             }}
             return json.dumps(error_json)
+    def extract_json(text):
+        start = text.find('{')
+        end = text.rfind('}') + 1
+        if start != -1 and end != -1:
+            json_str = text[start:end]
+            return json.loads(json_str)
+        raise ValueError("Valid JSON not found in text.")
 
 # Example usage (for testing outside Flask app context):
 if __name__ == '__main__':