Spaces:
Sleeping
Sleeping
File size: 3,008 Bytes
9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce 1fd9ee7 9d8abce |
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
from openai import OpenAI
import time
import re
# Streamlit page setup
st.set_page_config(page_title="documentaitest")
st.title("documentaitest")
st.caption("Chat with an Ai Assistant on your Documents")
# Sidebar for API key input
with st.sidebar:
OPENAI_API_KEY = st.text_input("Enter your C2 Group of Technologies Access Key", type="password")
if not OPENAI_API_KEY:
st.error("Please enter your C2 Group of Technologies Access Key to continue.")
st.stop()
client = OpenAI(api_key=OPENAI_API_KEY)
ASSISTANT_ID = "asst_PJjxQftfz2IJUUMvnldK58lB"
# Session state
if "messages" not in st.session_state:
st.session_state.messages = []
if "thread_id" not in st.session_state:
st.session_state.thread_id = None
# Clear chat
if st.button("Clear Chat", use_container_width=True):
st.session_state.messages = []
st.session_state.thread_id = None
st.rerun()
# Show history
for message in st.session_state.messages:
role, content = message["role"], message["content"]
st.chat_message(role).write(content)
# Handle new input
if prompt := st.chat_input():
st.session_state.messages.append({"role": "user", "content": prompt})
st.chat_message("user").write(prompt)
try:
if st.session_state.thread_id is None:
thread = client.beta.threads.create()
st.session_state.thread_id = thread.id
thread_id = st.session_state.thread_id
# Send user message
client.beta.threads.messages.create(
thread_id=thread_id,
role="user",
content=prompt
)
# Run assistant
run = client.beta.threads.runs.create(
thread_id=thread_id,
assistant_id=ASSISTANT_ID
)
# Poll until complete
while True:
run_status = client.beta.threads.runs.retrieve(thread_id=thread_id, run_id=run.id)
if run_status.status == "completed":
break
time.sleep(1)
# Get assistant response
messages = client.beta.threads.messages.list(thread_id=thread_id)
for message in reversed(messages.data):
if message.role == "assistant":
assistant_message = message.content[0].text.value
break
# Show response
st.chat_message("assistant").write(assistant_message)
st.session_state.messages.append({"role": "assistant", "content": assistant_message})
# 🔍 Detect and display GitHub-hosted image
page_match = re.search(r'Page (\d+)', assistant_message)
if page_match:
page_number = page_match.group(1).zfill(3)
image_url = f"https://raw.githubusercontent.com/AndrewLORTech/surgical-pathology-manual/main/51940670-Manual-of-Surgical-Pathology-Third-Edition_1_page_{page_number}.png"
st.image(image_url, caption=f"Page {int(page_number)}", use_column_width=True)
except Exception as e:
st.error(f"Error: {str(e)}")
|