Update app.py

app.py

@@ -3,25 +3,12 @@ import gradio as gr
 from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain import LLMChain, PromptTemplate
 from langchain.memory import ConversationBufferMemory
+from langchain.llms.google_genai import (
+    HarmBlockThreshold,
+    HarmCategory,
+)
+
 
-safety_settings = [
-  {
-    "category": "HARM_CATEGORY_HARASSMENT",
-    "threshold": "BLOCK_NONE"
-  },
-  {
-    "category": "HARM_CATEGORY_HATE_SPEECH",
-    "threshold": "BLOCK_NONE"
-  },
-  {
-    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
-    "threshold": "BLOCK_NONE"
-  },
-  {
-    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
-    "threshold": "BLOCK_NONE"
-  },
-]
 
 import getpass
 
@@ -50,7 +37,12 @@ memory = ConversationBufferMemory(memory_key="chat_history")
 
 llm_chain = LLMChain(
     llm=ChatGoogleGenerativeAI(model="gemini-1.5-flash-lastest"),
-    safety_settings=safety_settings,
+    safety_settings={
+            HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+            HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE,
+        },
     prompt=prompt,
     verbose=True,
     memory=memory,