import streamlit as st
import openai
import os

openai.api_key = "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"


def initialize_streamlit():
    st.set_page_config(initial_sidebar_state="collapsed")
    st.title("Asistent medical si oncologic - diagnostician")
    st.markdown(
        """
        Sunt un asistent oncologic cu vaste cunostinte in acest domeniu si expert in dignosticul diferentiat.
        """
    )

    if 'conversation' not in st.session_state:
        st.session_state.conversation = [{
            "role": "system", "content": """
                Tu esti un asistent foarte competent in domeniul oncologic si explici in detaliu atunci cand esti intrebat, cu 			   detalii stiintifice bine definite, iar atunci cand ti se cere parerea in cazuri pentru pacienti spui cele mai bune solutii, esti priceput un orice disciplina a medicinii si sti sa faci diagnostic diferentat pe orice tip de boala.
                """
        }]


def main():
    initialize_streamlit()
    question = st.text_input("Despre ce vrei sa vorbești?")
    if question == "": return
    st.session_state.conversation.append({"role": "user", "content": question})
    response = openai.ChatCompletion.create(
        model="gpt-3.5-turbo",
        messages=st.session_state.conversation
    )
    answer = response['choices'][0]['message']['content']
    st.session_state.conversation.append({"role": "assistant", "content": answer}, )
    st.write(answer)


main()