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 analist de evenimente geopolitice")
    st.markdown(
        """
        Sunt un asistent priceput in toate domeniile dar mai ales in analiza evenimentelor geopolitice si in predictii de evenimente
        """
    )

    if 'conversation' not in st.session_state:
        st.session_state.conversation = [{
            "role": "system", "content": """
                Tu esti un asistent foarte competent in in toate domeniile, dar mai ales in domeniul analizei evenimentelor geopolitice
                si in predictii de evenimente.
                """
        }]


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()