Upload app.py

app.py

@@ -588,62 +588,7 @@ SAFETY_SETTINGS = [
     {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": 0},
     {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": 0},
 ]
-pdf_mode = False
-# ------------------------------------------------------------
-if pdf_mode:
-    with st.sidebar:
-        st.title("Menu:")
-        pdf_docs = st.file_uploader("Upload your PDF Files and Click on the Submit & Process Button", accept_multiple_files=True)
-        if st.button("Submit & Process"):
-            with st.spinner("Processing..."):
-                raw_text = get_pdf_text(pdf_docs)
-                text_chunks = get_text_chunks(raw_text)
-                get_vector_store(text_chunks)
-                st.success("Done")
-
-    generation_config = {
-        "temperature": temperature,
-        "top_p": top_p,
-        "top_k": top_k,
-        "max_output_tokens": max_output_tokens,
-    }
-    safety_settings = SAFETY_SETTINGS
-else:
-    with st.sidebar:
-        st.subheader("Settings")
-        temperature1 = st.slider("Temperature", 0.1, 1.0, temperature)
-        top_p1 = st.slider("Top P", 0.1, 1.0, top_p, 0.01) 
-        top_k1 = st.slider("Top K", 1, 30, top_k)
-        max_output_tokens1 = st.slider("Max Output Tokens", 100, 2048, max_output_tokens)
 
-        st.subheader("Safety Settings")
-        safety_settings = []
-        for setting in SAFETY_SETTINGS:
-            category = setting["category"]
-            index = st.selectbox(f"{category} Threshold", range(len(threshold_options)))
-            
-            # Update safety settings based on user input
-            safety_settings.append({"category": category, "threshold": threshold_options[index]})
-
-        # Update generation_config based on user input
-        temperature = temperature1
-        top_p = top_p1
-        top_k = top_k1
-        max_output_tokens = max_output_tokens1
-        generation_config = {
-            "temperature": temperature,
-            "top_p": top_p,
-            "top_k": top_k,
-            "max_output_tokens": max_output_tokens,
-        }
-
-        st.sidebar.subheader("Selected Values")
-        st.sidebar.write(f"Temperature: {temperature}")
-        st.sidebar.write(f"Top P: {top_p}")
-        st.sidebar.write(f"Top K: {top_k}")
-        st.sidebar.write(f"Max Output Tokens: {max_output_tokens}")
-            
-#HEADER
 st.markdown('''
 Powered by Hk4crprasad <img src="https://i.ibb.co/BVJ2tCG/OIG-47-modified.png" width="20" height="20">
 , Streamlit and Python''', unsafe_allow_html=True)
@@ -699,7 +644,7 @@ def load_model() -> genai.GenerativeModel:
     'gemini-pro'.
     :return: an instance of the `genai.GenerativeModel` class.
     """
-    model = genai.GenerativeModel('gemini-pro', safety_settings=safety_settings, generation_config=generation_config)
+    model = genai.GenerativeModel('gemini-pro')
     return model
 
 @st.cache_resource
@@ -805,7 +750,45 @@ else:
     url = ''
 
 
+if pdf_mode:
+    with st.sidebar:
+        st.title("Menu:")
+        pdf_docs = st.file_uploader("Upload your PDF Files and Click on the Submit & Process Button", accept_multiple_files=True)
+        if st.button("Submit & Process"):
+            with st.spinner("Processing..."):
+                raw_text = get_pdf_text(pdf_docs)
+                text_chunks = get_text_chunks(raw_text)
+                get_vector_store(text_chunks)
+                st.success("Done")
+else:
+    with st.sidebar:
+        st.subheader("Settings")
+        temperature1 = st.slider("Temperature", 0.1, 1.0, temperature)
+        top_p1 = st.slider("Top P", 0.1, 1.0, top_p, 0.01) 
+        top_k1 = st.slider("Top K", 1, 30, top_k)
+        max_output_tokens1 = st.slider("Max Output Tokens", 100, 2048, max_output_tokens)
 
+        st.subheader("Safety Settings")
+        safety_settings = []
+        for setting in SAFETY_SETTINGS:
+            category = setting["category"]
+            index = st.selectbox(f"{category} Threshold", range(len(threshold_options)))
+            
+            # Update safety settings based on user input
+            safety_settings.append({"category": category, "threshold": threshold_options[index]})
+
+        # Update generation_config based on user input
+        temperature = temperature1
+        top_p = top_p1
+        top_k = top_k1
+        max_output_tokens = max_output_tokens1
+        st.sidebar.subheader("Selected Values")
+        st.sidebar.write(f"Temperature: {temperature}")
+        st.sidebar.write(f"Top P: {top_p}")
+        st.sidebar.write(f"Top K: {top_k}")
+        st.sidebar.write(f"Max Output Tokens: {max_output_tokens}")
+            
+#HEADER
 if txt_atachment:
     txtattachment = st.file_uploader("Upload your text file", type=['txt, py, go, java, php, etc'])
 else:
@@ -873,7 +856,17 @@ if prompt:
     
     with st.spinner(spinertxt):
         if len(prmt['parts']) > 1:
-            response = vision.generate_content(prmt['parts'],stream=True,safety_settings=safety_settings, generation_config=generation_config)
+            response = vision.generate_content(prmt['parts'],stream=True,safety_settings=[
+                    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
+                    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": threshold_options[index] for index in range(len(threshold_options))},
+                    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
+                    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
+                ], generation_config = {
+        "temperature": temperature,
+        "top_p": top_p,
+        "top_k": top_k,
+        "max_output_tokens": max_output_tokens,
+    })
             response.resolve()
         else:
             response = None
@@ -926,7 +919,17 @@ if prompt:
                 else:        
                     piro = f"{prmt['parts'][0]}"
         
-                st.session_state.chat.send_message(f"{piro}", stream=True)
+                st.session_state.chat.send_message(f"{piro}",generation_config = {
+                    "temperature": temperature,
+                    "top_p": top_p,
+                    "top_k": top_k,
+                    "max_output_tokens": max_output_tokens,
+                },safety_settings=[
+                    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
+                    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": threshold_options[index] for index in range(len(threshold_options))},
+                    {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
+                    {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": threshold_options[index] for index in range(len(threshold_options))},
+                ], stream=True)
                 response = st.session_state.chat.last
                 response.resolve()
             except StopCandidateException as sce: