added

app.py

@@ -97,7 +97,7 @@ def extract_products(text):
 
 def extract_with_perplexity_llm(ocr_text):
     prompt = f"""
-Extract the following fields from this receipt text and return them as a single flat JSON object:
+Extract the following fields from this receipt text and return them as a single flat JSON object (not a list or nested object):
 - name (customer name)
 - product (main product name)
 - price (main product price)
@@ -221,16 +221,16 @@ def main():
                     with st.spinner("Extracting structured data with Perplexity LLM..."):
                         llm_result = extract_with_perplexity_llm(pytesseract.image_to_string(Image.open(temp_path)))
                         st.subheader("Structured Data (Perplexity LLM)")
-                        st.code(llm_result, language="json")
-
-                        # Try to parse the JSON from the LLM output
-                        llm_data = {}
-                        try:
-                            llm_json = extract_json_from_llm_output(llm_result)
-                            if llm_json:
-                                llm_data = json.loads(llm_json)
-                        except Exception as e:
-                            st.error(f"Failed to parse LLM output as JSON: {e}")
+                        st.json(llm_result)
+
+                    # Try to parse the JSON from the LLM output
+                    llm_data = {}
+                    try:
+                        llm_json = extract_json_from_llm_output(llm_result)
+                        if llm_json:
+                            llm_data = json.loads(llm_json)
+                    except Exception as e:
+                        st.error(f"Failed to parse LLM output as JSON: {e}")
 
                     # Merge results
                     final_data = merge_extractions(fields, llm_data)