save

App/Chat/utils/Summarize.py

@@ -6,9 +6,10 @@ from langchain import PromptTemplate
 import os
 from poe_api_wrapper import PoeApi
 import pprint
+
 client = PoeApi("sXvCnfYy8CHnXNTRlxhmVg==")
 bot = "Assistant"
-CHAT_CODE=''
+CHAT_CODE = ""
 
 PALM_API = ""
 API_KEY = os.environ.get("PALM_API", PALM_API)
@@ -16,7 +17,10 @@ palm.configure(api_key=API_KEY)
 
 
 text_splitter = RecursiveCharacterTextSplitter(
-    separators=["\n\n", "\n", "."], chunk_size=1500,length_function = len, chunk_overlap=100
+    separators=["\n\n", "\n", "."],
+    chunk_size=1500,
+    length_function=len,
+    chunk_overlap=100,
 )
 
 
@@ -80,16 +84,17 @@ def count_tokens(text):
 
 async def PalmTextModel(message):
     global CHAT_CODE
-    if CHAT_CODE == '':
+    if CHAT_CODE == "":
         for chunk in client.send_message(bot, message):
             pass
-        CHAT_CODE=chunk['chatCode']
+        CHAT_CODE = chunk["chatCode"]
     else:
-        for chunk in client.send_message(bot, message,chatCode=CHAT_CODE):
+        for chunk in client.send_message(bot, message, chatCode=CHAT_CODE):
             pass
 
     return chunk["text"]
 
+
 async def Summarizer(essay):
     docs = text_splitter.create_documents([essay])