from openai import OpenAI
import streamlit as st
from groq_response import groq_response
import os

    

client = OpenAI(api_key=os.getenv('OPENAI_API'))
prompt = st.chat_input("Type your question...")
my_assistant = client.beta.assistants.retrieve("asst_ucgc8lzekHeOFrvjaKzcJOmD")
ass_id = "asst_ucgc8lzekHeOFrvjaKzcJOmD"

def retrival_openai(prompt, instructions):
    thread = client.beta.threads.create()
    message = client.beta.threads.messages.create(
        thread_id=thread.id,
        role="user",
        content=prompt
    )
    run = client.beta.threads.runs.create_and_poll(
        thread_id=thread.id,
        assistant_id=ass_id,
        instructions=instructions
    )
    if run.status == 'completed': 
        messages = client.beta.threads.messages.list(
            thread_id=thread.id
        )
        assistant_messages = [message for message in messages.data if message.role == 'assistant']
        for message in assistant_messages:
            for content_block in message.content:
                # Check if there's a method or property to get 'value'
                if hasattr(content_block.text, 'value'):
                    return content_block.text.value                    
                else:
                    # Otherwise print the object to debug
                    #st.write(messages)
                    #st.write(content_block.text)
                    return None
    else:
        return None