Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import os
 import time
 import json
+import re
 from openai import OpenAI
 
 # Configuration
@@ -67,19 +68,20 @@ if st.session_state.tech_messages and st.session_state.tech_messages[-1]["role"]
                         reply_content = message.content[0].text.value.strip()
                         st.session_state.tech_messages.append({"role": "assistant", "content": reply_content})
 
-                        # Try to parse response as JSON
+                        # Try to parse response as JSON block (```json ... ```)
                         try:
-                            results = json.loads(reply_content)
+                            match = re.search(r"```json\s*(.*?)```", reply_content, re.DOTALL)
+                            json_str = match.group(1) if match else reply_content
+                            results = json.loads(json_str)
+
                             if isinstance(results, list):
                                 for item in results:
                                     st.markdown(f"### 📄 {item.get('drawing_number')} ({item.get('discipline')})")
                                     st.markdown(f"**Summary:** {item.get('summary')}")
 
-                                    # Single image (keyword match)
                                     if "image" in item:
                                         st.image(item["image"], use_column_width=True)
 
-                                    # Multiple image pages
                                     elif "images" in item:
                                         for img in item["images"]:
                                             st.image(img, use_column_width=True)