Update app.py

app.py

@@ -134,14 +134,21 @@ This JSON format will be used to store and manage invoices in a structured and u
             max_tokens=16384
         )
         content = response.choices[0].message.content.strip()
-        #cleaned_content = re.sub(r'^```json\n', '', content)  # Remove '```json\n' at the beginning
-        #cleaned_content = re.sub(r'\n```$', '', cleaned_content)  # Remove '\n```' at the end
+        cleaned_content = re.sub(r'^```json\n', '', content)  # Remove '```json\n' at the beginning
+        cleaned_content = re.sub(r'\n```$', '', cleaned_content)  # Remove '\n```' at the end
 
         # Step 2: Parse the cleaned content as JSON
         #parsed_content = json.loads(cleaned_content)
 
         # Step 3: Print the parsed JSON object
-        return content # parsed_content
+        try:
+            parsed_content = json.loads(cleaned_content)
+            return parsed_content
+        except json.JSONDecodeError as e:
+            print("Error parsing JSON:", e)
+            # Optionally, print the cleaned content to debug
+            print("Cleaned content:", cleaned_content)
+            return None
     except Exception as e:
         return f"Error in summarization: {str(e)}"
 # Dependency to check API Key