Update app.py

app.py

@@ -10,7 +10,7 @@ from openai import OpenAI
 # ------------------ App Configuration ------------------
 st.set_page_config(page_title="Document AI Assistant", layout="wide")
 st.title("📄 Document AI Assistant")
-st.caption("Chat with an AI Assistant on your medical/pathology or construction documents")
+st.caption("Chat with an AI Assistant on your construction documents")
 
 # ------------------ Load API Key and Assistant ID ------------------
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
@@ -22,19 +22,6 @@ if not OPENAI_API_KEY or not ASSISTANT_ID:
 
 client = OpenAI(api_key=OPENAI_API_KEY)
 
-# ------------------ Map document titles to GitHub folder names ------------------
-DOCUMENT_FOLDER_MAP = {
-    "Tender 5 of 2024 - Forrestdale": "Schlager Group Pty Ltd - Tender 5 of 2024 - Forrestdale ~ Pavilion and External Works Part 1 of 2",
-    "Tender 5 of 2024 - Forrestdale Part 2": "Schlager Group Pty Ltd - Tender 5 of 2024 - Forrestdale ~ Pavilion and External Works Part 2 of 2",
-    "Armadale Forrestdale Methodology": "Armadale Forrestdale Methodology",
-    "2302-FORRESTDALE SPORTING PAVILION-TECHNICAL SPECIFICATION": "2302-FORRESTDALE SPORTING PAVILION-TECHNICAL SPECIFICATION",
-    "22.146.DS Electrical Specification IFC Rev_0": "22.146.DS Electrical Specification IFC Rev_0",
-    "Forrrestdale Community Hub Program 23-04-2025": "Forrrestdale Community Hub Program 23-04-2025",
-    "Landscape Specification - Forrestdale Community Hub_02": "Landscape Specification - Forrestdale Community Hub_02",
-    "Mechanical_Spec_-_Forrestdale Sporting Club Room IFC": "Mechanical_Spec_-_Forrestdale Sporting Club Room IFC",
-    "V-23-021-HY-SPEC-R0": "V-23-021-HY-SPEC-R0"
-}
-
 # ------------------ Session State Initialization ------------------
 if "messages" not in st.session_state:
     st.session_state.messages = []
@@ -123,7 +110,7 @@ with col2:
                         break
                     time.sleep(1)
 
-            # Retrieve assistant message
+            # Retrieve assistant reply
             messages = client.beta.threads.messages.list(thread_id=st.session_state.thread_id)
             assistant_message = None
             for message in reversed(messages.data):
@@ -133,21 +120,16 @@ with col2:
 
             st.session_state.messages.append({"role": "assistant", "content": assistant_message})
 
-            # ✅ Extract folder + page number from "Document Reference" line
+            # ✅ Parse Document Reference line and generate correct image URL
             match = re.search(r'Document Reference:\s+(.+?),\s+Page\s+(\d+)', assistant_message)
             if match:
-                doc_name_raw = match.group(1).strip()
-                page_num = int(match.group(2))
-                page_str = f"{page_num:04d}"
-
-                # Map to actual GitHub folder name
-                folder_name = DOCUMENT_FOLDER_MAP.get(doc_name_raw, doc_name_raw)
-
+                doc_name = match.group(1).strip()
+                page = int(match.group(2))
+                page_str = f"{page:04d}"
                 corrected_url = (
                     f"https://raw.githubusercontent.com/AndrewLORTech/c2ozschlaegerforrestdale/main/"
-                    f"{folder_name}/{folder_name}_page_{page_str}.png"
+                    f"{doc_name}/{doc_name}_page_{page_str}.png"
                 )
-
                 st.session_state.image_url = corrected_url
                 st.session_state.image_updated = True