Update app.py

app.py

@@ -1,108 +1,59 @@
-import re
-from pathlib import Path
-import pickle
 
 import streamlit as st
-import openai
-from unidecode import unidecode
-
-PASTA_CONFIGURACOES = Path(__file__).parent / 'configuracoes'
-PASTA_CONFIGURACOES.mkdir(exist_ok=True)
-PASTA_MENSAGENS = Path(__file__).parent / 'mensagens'
-PASTA_MENSAGENS.mkdir(exist_ok=True)
-CACHE_DESCONVERTE = {}
-
-# API OPENAI ================================================
-def retorna_resposta_modelo(mensagens,
-                            openai_key,
-                            modelo='gpt-4o-mini',
-                            temperatura=0,
-                            stream=False):
-    openai.api_key = openai_key
-    response = openai.ChatCompletion.create(
-        model=modelo,
-        messages=mensagens,
-        temperature=temperatura,
-        stream=stream
-    )
-    return response
-
-# SALVAMENTO E LEITURA DE CONVERSAS ========================
-
-def converte_nome_mensagem(nome_mensagem):
-    nome_arquivo = unidecode(nome_mensagem)
-    nome_arquivo = re.sub('\\W+', '', nome_arquivo).lower()
-    return nome_arquivo
-
-def desconverte_nome_mensagem(nome_arquivo):
-    if not nome_arquivo in CACHE_DESCONVERTE:
-        nome_mensagem = ler_mensagem_por_nome_arquivo(nome_arquivo, key='nome_mensagem')
-        CACHE_DESCONVERTE[nome_arquivo] = nome_mensagem
-    return CACHE_DESCONVERTE[nome_arquivo]
-
-def retorna_nome_da_mensagem(mensagens):
-    nome_mensagem = ''
-    for mensagem in mensagens:
-        if mensagem['role'] == 'user':
-            nome_mensagem = mensagem['content'][:30]
-            break
-    return nome_mensagem
-
-def salvar_mensagens(mensagens):
-    if len(mensagens) == 0:
-        return False
-    nome_mensagem = retorna_nome_da_mensagem(mensagens)
-    nome_arquivo = converte_nome_mensagem(nome_mensagem)
-    arquivo_salvar = {'nome_mensagem': nome_mensagem,
-                      'nome_arquivo': nome_arquivo,
-                      'mensagem': mensagens}
-    with open(PASTA_MENSAGENS / nome_arquivo, 'wb') as f:
-        pickle.dump(arquivo_salvar, f)
-
-def ler_mensagem_por_nome_arquivo(nome_arquivo, key='mensagem'):
-    with open(PASTA_MENSAGENS / nome_arquivo, 'rb') as f:
-        mensagens = pickle.load(f)
-    return mensagens[key]
-
-def ler_mensagens(mensagens, key='mensagem'):
-    if len(mensagens) == 0:
-        return []
-    nome_mensagem = retorna_nome_da_mensagem(mensagens)
-    nome_arquivo = converte_nome_mensagem(nome_mensagem)
-    with open(PASTA_MENSAGENS / nome_arquivo, 'rb') as f:
-        mensagens = pickle.load(f)
-    return mensagens[key]
-
-def listar_conversas():
-    conversas = list(PASTA_MENSAGENS.glob('*'))
-    conversas = sorted(conversas, key=lambda item: item.stat().st_mtime_ns, reverse=True)
-    return [c.stem for c in conversas]
-
-# SALVAMENTO E LEITURA DA APIKEY ========================
-
-def salva_chave(chave):
-    with open(PASTA_CONFIGURACOES / 'chave', 'wb') as f:
-        pickle.dump(chave, f)
-
-def le_chave():
-    if (PASTA_CONFIGURACOES / 'chave').exists():
-        with open(PASTA_CONFIGURACOES / 'chave', 'rb') as f:
-            return pickle.load(f)
-    else:
-        return ''
 
-# PÁGINAS ==================================================
+from utils_openai import retorna_resposta_modelo
+from utils_files import *
 
+# 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-4o-mini'
+        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'])
@@ -143,43 +94,7 @@ def pagina_principal():
             st.session_state['mensagens'] = mensagens
             salvar_mensagens(mensagens)
 
-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-4o-mini','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')
-
+# MAIN ==================================================
 def main():
     inicializacao()
     pagina_principal()
@@ -188,6 +103,5 @@ def main():
     tab_configuracoes(tab2)
 
 
-
 if __name__ == '__main__':
     main()