Update app.py

app.py

@@ -1,92 +1,16 @@
+import base64
+import requests
+import tempfile
 import streamlit as st
 from openai import OpenAI
-import time
-import re
 
-st.set_page_config(page_title="documentaitest")
-st.title("documentaitest")
-st.caption("Chat with an Ai Assistant on your Documents")
-
-# Sidebar key entry
-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 setup
-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 button
-if st.button("Clear Chat", use_container_width=True):
-    st.session_state.messages = []
-    st.session_state.thread_id = None
-    st.rerun()
-
-# Display prior messages
-for message in st.session_state.messages:
-    role, content = message["role"], message["content"]
-    st.chat_message(role).write(content)
-
-# Handle 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 prompt to assistant
-        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
+# Whisper transcription function
+def transcribe_audio(file_path, api_key):
+    with open(file_path, "rb") as f:
+        response = requests.post(
+            "https://api.openai.com/v1/audio/transcriptions",
+            headers={"Authorization": f"Bearer {api_key}"},
+            files={"file": f},
+            data={"model": "whisper-1"}
         )
-
-        # Wait for assistant to finish
-        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)
-        assistant_message = None
-        for message in reversed(messages.data):
-            if message.role == "assistant":
-                assistant_message = message.content[0].text.value
-                break
-
-        # Display message
-        st.chat_message("assistant").write(assistant_message)
-        st.session_state.messages.append({"role": "assistant", "content": assistant_message})
-
-        # 🔍 Extract and display GitHub image if present
-        image_match = re.search(r'https://raw\.githubusercontent\.com/AndrewLORTech/surgical-pathology-manual/main/[\w\-/]*\.png', assistant_message)
-        if image_match:
-            image_url = image_match.group(0)
-            st.image(image_url, caption="Page Image", use_container_width=True)
-
-    except Exception as e:
-        st.error(f"Error: {str(e)}")
+    return response.json().get("text", None)