Update app.py

app.py

@@ -2,6 +2,7 @@ import streamlit as st
 import requests
 import os
 import google.generativeai as genai
+from google.generativeai.types import HarmCategory, HarmBlockThreshold
 
 genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))  
 
@@ -44,7 +45,13 @@ def get_summary(text):
                    Other opinions (if there are any dissenting or concurring opinions, summarize majority opinion, dissenting opinion and concurring opinion).
                    Cases cited (which cases the court cited and how it treated them).
                    Here is the text of the case to be summarized: {text}''',
-            stream=False 
+            stream=False, 
+            safety_settings={
+        HarmCategory.HARM_CATEGORY_HATE_SPEECH: HarmBlockThreshold.BLOCK_NONE,
+        HarmCategory.HARM_CATEGORY_HARASSMENT: HarmBlockThreshold.BLOCK_NONE,
+                HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: HarmBlockThreshold.BLOCK_NONE,
+                HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: HarmBlockThreshold.BLOCK_NONE
+            }
         )
         return response 
     except Exception as e: