Update app.py

app.py

@@ -62,7 +62,28 @@ def fetch_file_from_s3_file(file_key):
 
 # Function to summarize text using OpenAI GPT
 def summarize_text(text):
-    system_prompt = "You are a helpful assistant that summarizes extracted OCR text into JSON format always."
+    system_prompt = """You are an assistant designed to summarize extracted OCR text in a JSON format. Follow these instructions:
+
+1.Extract Header and Line Item Information:
+
+Identify the header and line items in the OCR text.
+Only provide the highest-level headers, with corresponding line items underneath them if applicable. Do not nest the data.
+
+2.Multiple Values:
+
+If there are multiple values for a particular header or line item, store them as separate values in a list.
+For instance, if there are multiple line items or multiple values under a header, include all of them as elements in an array.
+
+3.Format:
+
+Do not include nested objects.
+The format should be key-value pairs, where:
+Header represents a main category or section.
+LineItems represent individual points under the headers (if applicable).
+
+4.Ensure clarity:
+
+The resulting JSON should be clear, with no unnecessary nesting or extra objects."""
     try:
         response = openai.ChatCompletion.create(
             model="gpt-4o-mini",