Update app.py

app.py

@@ -35,6 +35,30 @@ DUPLICATE = """
 
 IMAGE_WIDTH = 512
 
+
+afety_settings = [
+    {
+        "category": "HARM_CATEGORY_DANGEROUS",
+        "threshold": "BLOCK_NONE",
+    },
+    {
+        "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",
+    },]
+
+
 def extract_text_from(vid_link):
   video_url = vid_link
   yt = YouTube(video_url)
@@ -94,12 +118,13 @@ def bot(
         text_prompt = txt_in
 
     genai.configure(api_key=google_key)
+
     generation_config = genai.types.GenerationConfig(
-        temperature=temperature,
-        max_output_tokens=max_output_tokens,
+        temperature=0.7,
+        max_output_tokens=2048
         stop_sequences=preprocess_stop_sequences(stop_sequences=stop_sequences),
-        top_k=top_k,
-        top_p=top_p)
+        top_k=40,
+        top_p=0.95)
 
     if image_prompt is None:
         prompt= "استخرج كلمات مفتاحية من النص التالي: "+text_prompt
@@ -107,13 +132,13 @@ def bot(
         response = model.generate_content(
             prompt,
             stream=True,
-            generation_config=generation_config)
+            generation_config=generation_config,afety_settings=afety_settings)
         out1 = response.text
         prompt = "أذكر لي آية من القران الكريم تتحدث عن أحد هذه المواضيع او اكثر: "+ out1 + " واشرح الآيه وفسرها باللغة العربية."
         response = model.generate_content(
             prompt,
             stream=True,
-            generation_config=generation_config)
+            generation_config=generation_config, afety_settings=afety_settings)
         response.resolve()
     else:
         image_prompt = preprocess_image(image_prompt)
@@ -121,7 +146,7 @@ def bot(
         response = model.generate_content(
             contents=[text_prompt, image_prompt],
             stream=True,
-            generation_config=generation_config)
+            generation_config=generation_config, afety_settings=afety_settings)
         response.resolve()
 
     # streaming effect