Update app.py

app.py

@@ -2,6 +2,9 @@ import streamlit as st
 import os
 import time
 import re
+import requests
+from PIL import Image
+from io import BytesIO
 from openai import OpenAI
 
 # ------------------ App Configuration ------------------
@@ -14,7 +17,7 @@ OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 ASSISTANT_ID = os.environ.get("ASSISTANT_ID")
 
 if not OPENAI_API_KEY or not ASSISTANT_ID:
-    st.error("❌ Missing secrets. Please set both OPENAI_API_KEY and ASSISTANT_ID in Hugging Face Space secrets.")
+    st.error("❌ Missing secrets. Please set both OPENAI_API_KEY and ASSISTANT_ID in your Hugging Face Space secrets.")
     st.stop()
 
 client = OpenAI(api_key=OPENAI_API_KEY)
@@ -46,8 +49,14 @@ col1, col2 = st.columns([1, 2])
 # ------------------ Left Panel: Image ------------------
 with col1:
     if show_image and st.session_state.image_url:
-        st.image(st.session_state.image_url, caption="📑 Extracted Page", use_container_width=True)
-        st.session_state.image_updated = False
+        try:
+            response = requests.get(st.session_state.image_url)
+            response.raise_for_status()
+            img = Image.open(BytesIO(response.content))
+            st.image(img, caption="📑 Extracted Page", use_container_width=True)
+            st.session_state.image_updated = False
+        except Exception as e:
+            st.warning(f"⚠️ Failed to load image from URL:\n{st.session_state.image_url}\n\nError: {e}")
 
 # ------------------ Right Panel: Chat ------------------
 with col2:
@@ -78,7 +87,7 @@ with col2:
                 thread = client.beta.threads.create()
                 st.session_state.thread_id = thread.id
 
-            # Send message to assistant
+            # Send user prompt
             client.beta.threads.messages.create(
                 thread_id=st.session_state.thread_id,
                 role="user",
@@ -112,7 +121,7 @@ with col2:
 
             st.session_state.messages.append({"role": "assistant", "content": assistant_message})
 
-            # ✅ Updated: Extract any GitHub-hosted .png image URL (flexible across repos)
+            # ✅ Extract GitHub raw image URL
             image_match = re.search(
                 r'https://raw\.githubusercontent\.com/AndrewLORTech/[a-zA-Z0-9_\-/]+\.png',
                 assistant_message