Summarize this

App/Chat/utils/Summarize.py

@@ -45,56 +45,56 @@ def count_tokens(text):
     return palm.count_message_tokens(prompt=text)["token_count"]
 
 
-# async def PalmTextModel(text, candidates=1):
-#     url = f"https://generativelanguage.googleapis.com/v1beta2/models/text-bison-001:generateText?key={API_KEY}"
-
-#     headers = {
-#         "Content-Type": "application/json",
-#     }
-
-#     data = {
-#         "prompt": {"text": text},
-#         "temperature": 0.95,
-#         "top_k": 100,
-#         "top_p": 0.95,
-#         "candidate_count": candidates,
-#         "max_output_tokens": 1024,
-#         "stop_sequences": ["</output>"],
-#         "safety_settings": [
-#             {"category": "HARM_CATEGORY_DEROGATORY", "threshold": 4},
-#             {"category": "HARM_CATEGORY_TOXICITY", "threshold": 4},
-#             {"category": "HARM_CATEGORY_VIOLENCE", "threshold": 4},
-#             {"category": "HARM_CATEGORY_SEXUAL", "threshold": 4},
-#             {"category": "HARM_CATEGORY_MEDICAL", "threshold": 4},
-#             {"category": "HARM_CATEGORY_DANGEROUS", "threshold": 4},
-#         ],
-#     }
-
-#     async with aiohttp.ClientSession() as session:
-#         async with session.post(url, json=data, headers=headers) as response:
-#             if response.status == 200:
-#                 result = await response.json()
-#                 # print(result)
-#                 if candidates > 1:
-#                     temp = [candidate["output"] for candidate in result["candidates"]]
-#                     return temp
-#                 temp = result["candidates"][0]["output"]
-#                 return temp
-#             else:
-#                 print(f"Error: {response.status}\n{await response.text()}")
-
-
-async def PalmTextModel(message):
-    global CHAT_CODE
-    if CHAT_CODE == "":
-        for chunk in client.send_message(bot, message):
-            pass
-        CHAT_CODE = chunk["chatCode"]
-    else:
-        for chunk in client.send_message(bot, message, chatCode=CHAT_CODE):
-            pass
-
-    return chunk["text"]
+async def PalmTextModel(text, candidates=1):
+    url = f"https://generativelanguage.googleapis.com/v1beta2/models/text-bison-001:generateText?key={API_KEY}"
+
+    headers = {
+        "Content-Type": "application/json",
+    }
+
+    data = {
+        "prompt": {"text": text},
+        "temperature": 0.95,
+        "top_k": 100,
+        "top_p": 0.95,
+        "candidate_count": candidates,
+        "max_output_tokens": 1024,
+        "stop_sequences": ["</output>"],
+        "safety_settings": [
+            {"category": "HARM_CATEGORY_DEROGATORY", "threshold": 4},
+            {"category": "HARM_CATEGORY_TOXICITY", "threshold": 4},
+            {"category": "HARM_CATEGORY_VIOLENCE", "threshold": 4},
+            {"category": "HARM_CATEGORY_SEXUAL", "threshold": 4},
+            {"category": "HARM_CATEGORY_MEDICAL", "threshold": 4},
+            {"category": "HARM_CATEGORY_DANGEROUS", "threshold": 4},
+        ],
+    }
+
+    async with aiohttp.ClientSession() as session:
+        async with session.post(url, json=data, headers=headers) as response:
+            if response.status == 200:
+                result = await response.json()
+                # print(result)
+                if candidates > 1:
+                    temp = [candidate["output"] for candidate in result["candidates"]]
+                    return temp
+                temp = result["candidates"][0]["output"]
+                return temp
+            else:
+                print(f"Error: {response.status}\n{await response.text()}")
+
+
+# async def PalmTextModel(message):
+#     global CHAT_CODE
+#     if CHAT_CODE == "":
+#         for chunk in client.send_message(bot, message):
+#             pass
+#         CHAT_CODE = chunk["chatCode"]
+#     else:
+#         for chunk in client.send_message(bot, message, chatCode=CHAT_CODE):
+#             pass
+
+#     return chunk["text"]
 
 
 async def Summarizer(essay):