Spaces:
Sleeping
Sleeping
File size: 3,672 Bytes
9d10d92 13919c8 930629f 13919c8 b386f62 930629f c1043ca 8c4492e 930629f c1043ca 930629f 8c4492e c1043ca 8c4492e c1043ca 930629f 2e0eb4d 930629f 56b7b50 930629f 2e0eb4d 930629f 2e0eb4d 930629f 56b7b50 2e0eb4d 56b7b50 2e0eb4d 56b7b50 2e0eb4d 930629f 56b7b50 930629f 56b7b50 930629f 56b7b50 2e0eb4d 56b7b50 930629f 56b7b50 930629f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
import streamlit as st
import os
import time
import json
from openai import OpenAI
# Configuration
st.set_page_config(page_title="Forrestdale Drawing Viewer", layout="wide")
st.title("📐 Forrestdale Technical Drawing Assistant")
OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
ASSISTANT_ID = "asst_DjvuWBc7tCvMbAhY7n1em4BZ"
if not OPENAI_API_KEY:
st.error("❌ Missing OPENAI_API_KEY.")
st.stop()
client = OpenAI(api_key=OPENAI_API_KEY)
if "tech_thread_id" not in st.session_state:
thread = client.beta.threads.create()
st.session_state.tech_thread_id = thread.id
if "tech_messages" not in st.session_state:
st.session_state.tech_messages = []
# Prompt input
prompt = st.chat_input("Ask about plans, drawings or components (e.g. Show me all electrical plans)")
if prompt:
st.session_state.tech_messages.append({"role": "user", "content": prompt})
# Display assistant messages
for msg in st.session_state.tech_messages:
with st.chat_message(msg["role"]):
st.markdown(msg["content"])
# Trigger assistant
if st.session_state.tech_messages and st.session_state.tech_messages[-1]["role"] == "user":
with st.spinner("⏳ Fetching results from assistant..."):
try:
client.beta.threads.messages.create(
thread_id=st.session_state.tech_thread_id,
role="user",
content=st.session_state.tech_messages[-1]["content"]
)
run = client.beta.threads.runs.create(
thread_id=st.session_state.tech_thread_id,
assistant_id=ASSISTANT_ID
)
while True:
run_status = client.beta.threads.runs.retrieve(
thread_id=st.session_state.tech_thread_id,
run_id=run.id
)
if run_status.status in ["completed", "failed", "cancelled"]:
break
time.sleep(1)
if run_status.status != "completed":
st.error("⚠️ Assistant run failed.")
else:
messages = client.beta.threads.messages.list(thread_id=st.session_state.tech_thread_id)
for message in reversed(messages.data):
if message.role == "assistant":
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:
results = json.loads(reply_content)
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)
except Exception as parse_error:
st.warning("🟡 Could not parse assistant response as JSON.")
break
except Exception as e:
st.error(f"❌ Error occurred: {e}")
|