Update app.py

app.py

@@ -1,64 +1,36 @@
 
 import streamlit as st
+import openai
+import dotenv
+import os
+
+
+def carregar_configuracoes():
+    dotenv.load_dotenv()
+    return os.environ['OPENAI_API_KEY']
+
+def retorna_resposta_modelo(mensagens,
+                            modelo='gpt-4o-mini',
+                            temperatura=0,
+                            stream=False):
+    openai.api_key = carregar_configuracoes()
+    response = openai.ChatCompletion.create(
+        model=modelo,
+        messages=mensagens,
+        temperature=temperatura,
+        stream=stream
+    )
+    return response
 
-from utils_openai import retorna_resposta_modelo
-from utils_files import *
+def pagina_principal():
 
-# INICIALIZAÇÃO ==================================================
-def inicializacao():
     if not 'mensagens' in st.session_state:
         st.session_state.mensagens = []
-    if not 'conversa_atual' in st.session_state:
-        st.session_state.conversa_atual = ''
-    if not 'modelo' in st.session_state:
-        st.session_state.modelo = 'gpt-3.5-turbo'
-    if not 'api_key' in st.session_state:
-        st.session_state.api_key = le_chave()
-
-# TABS ==================================================
-def tab_conversas(tab):
-
-    tab.button('➕ Nova conversa',
-                on_click=seleciona_conversa,
-                args=('', ),
-                use_container_width=True)
-    tab.markdown('')
-    conversas = listar_conversas()
-    for nome_arquivo in conversas:
-        nome_mensagem = desconverte_nome_mensagem(nome_arquivo).capitalize()
-        if len(nome_mensagem) == 30:
-            nome_mensagem += '...'
-        tab.button(nome_mensagem,
-            on_click=seleciona_conversa,
-            args=(nome_arquivo, ),
-            disabled=nome_arquivo==st.session_state['conversa_atual'],
-            use_container_width=True)
-
-def seleciona_conversa(nome_arquivo):
-    if nome_arquivo == '':
-        st.session_state['mensagens'] = []
-    else:
-        mensagem = ler_mensagem_por_nome_arquivo(nome_arquivo)
-        st.session_state['mensagens'] = mensagem
-    st.session_state['conversa_atual'] = nome_arquivo
-
-def tab_configuracoes(tab):
-    modelo_escolhido = tab.selectbox('Selecione o modelo',
-                                     ['gpt-3.5-turbo', 'gpt-4'])
-    st.session_state['modelo'] = modelo_escolhido
-
-    chave = tab.text_input('Adicione sua api key', value=st.session_state['api_key'])
-    if chave != st.session_state['api_key']:
-        st.session_state['api_key'] = chave
-        salva_chave(chave)
-        tab.success('Chave salva com sucesso')
-
-# PÁGINA PRINCIPAL ==================================================
-def pagina_principal():
     
-    mensagens = ler_mensagens(st.session_state['mensagens'])
+    mensagens = st.session_state['mensagens']
 
-    st.header('🤖 Chatbot', divider=True)
+    st.markdown("<h2 style='text-align: center; margin-bottom: -100px; margin-top: -50px'>🤖 Chatbot</h2>", unsafe_allow_html=True)
+    st.write("---")
 
     for mensagem in mensagens:
         chat = st.chat_message(mensagem['role'])
@@ -66,42 +38,30 @@ def pagina_principal():
     
     prompt = st.chat_input('Fale com o chat')
     if prompt:
-        if st.session_state['api_key'] == '':
-            st.error('Adicone uma chave de api na aba de configurações')
-        else:
-            nova_mensagem = {'role': 'user',
-                            'content': prompt}
-            chat = st.chat_message(nova_mensagem['role'])
-            chat.markdown(nova_mensagem['content'])
-            mensagens.append(nova_mensagem)
+        nova_mensagem = {'role': 'user',
+                         'content': prompt}
+        chat = st.chat_message(nova_mensagem['role'])
+        chat.markdown(nova_mensagem['content'])
+        mensagens.append(nova_mensagem)
+
+        chat = st.chat_message('assistant')
+        placeholder = chat.empty()
+        placeholder.markdown("▌")
+        resposta_completa = ''
+        respostas = retorna_resposta_modelo(mensagens,
+                                            stream=True)
+        for resposta in respostas:
+            resposta_completa += resposta.choices[0].delta.get('content', '')
+            placeholder.markdown(resposta_completa + "▌")
+        placeholder.markdown(resposta_completa)
+        nova_mensagem = {'role': 'assistant',
+                         'content': resposta_completa}
+        mensagens.append(nova_mensagem)
+
 
-            chat = st.chat_message('assistant')
-            placeholder = chat.empty()
-            placeholder.markdown("▌")
-            resposta_completa = ''
-            respostas = retorna_resposta_modelo(mensagens,
-                                                st.session_state['api_key'],
-                                                modelo=st.session_state['modelo'],
-                                                stream=True)
-            for resposta in respostas:
-                resposta_completa += resposta.choices[0].delta.get('content', '')
-                placeholder.markdown(resposta_completa + "▌")
-            placeholder.markdown(resposta_completa)
-            nova_mensagem = {'role': 'assistant',
-                            'content': resposta_completa}
-            mensagens.append(nova_mensagem)
+        st.session_state['mensagens'] = mensagens
 
-            st.session_state['mensagens'] = mensagens
-            salvar_mensagens(mensagens)
 
-# MAIN ==================================================
-def main():
-    inicializacao()
-    pagina_principal()
-    tab1, tab2 = st.sidebar.tabs(['Conversas', 'Configurações'])
-    tab_conversas(tab1)
-    tab_configuracoes(tab2)
 
 
-if __name__ == '__main__':
-    main()
+pagina_principal()