Upload app.py

app.py

@@ -577,31 +577,19 @@ st.set_page_config(
 )
 # Path: Main.py
 #Author: Sergio Demis Lopez Martinez
-
 temperature = 0.9
 top_p = 1.0
 top_k = 1
 max_output_tokens = 1024
 threshold_options = ["BLOCK_NONE", "BLOCK_ONLY_HIGH", "BLOCK_MEDIUM_AND_ABOVE", "BLOCK_LOW_AND_ABOVE"]
-
-cols=st.columns(5)
-
-with cols[0]:
-    image_atachment = st.toggle("Attach image", value=False, help="Activate this mode to attach an image and let the chatbot read it")
-
-with cols[1]:
-    txt_atachment = st.toggle("Attach text file", value=False, help="Activate this mode to attach a text file and let the chatbot read it")
-    
-with cols[2]:
-    csv_excel_atachment = st.toggle("Attach CSV or Excel", value=False, help="Activate this mode to attach a CSV or Excel file and let the chatbot read it")
-    
-with cols[3]:
-    graphviz_mode = st.toggle("Graphviz mode", value=False, help="Activate this mode to generate a graph with graphviz in .dot from your message")
-    
-with cols[4]:
-    pdf_mode = st.toggle("Pdf ask mode", value=False, help="Activate this mode to generate a graph with graphviz in .dot from your message")
-#------------------------------------------------------------
-
+SAFETY_SETTINGS = [
+    {"category": "HARM_CATEGORY_HARASSMENT", "threshold": 0},  
+    {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": 0},
+    {"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:")
@@ -612,47 +600,31 @@ if pdf_mode:
                 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 = [
-        {"category": "HARM_CATEGORY_HARASSMENT", "threshold": 0},  # Default to BLOCK_NONE
-        {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": 0},
-        {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": 0},
-        {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": 0},
-    ]
+    safety_settings = SAFETY_SETTINGS
 else:
-    SAFETY_SETTINGS = [
-        {"category": "HARM_CATEGORY_HARASSMENT", "threshold": 0},  # Default to BLOCK_NONE
-        {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": 0},
-        {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": 0},
-        {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": 0},
-    ]
-    generation_config = {
-        "temperature": temperature,
-        "top_p": top_p,
-        "top_k": top_k,
-        "max_output_tokens": max_output_tokens,
-    }
     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)  # Use float step
+        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
-            setting["threshold"] = threshold_options[index]
-            
+            safety_settings.append({"category": category, "threshold": threshold_options[index]})
+
         # Update generation_config based on user input
         temperature = temperature1
         top_p = top_p1
@@ -664,20 +636,12 @@ else:
             "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}")
-        
-        SAFETY_SETTINGS = [
-            {"category": "HARM_CATEGORY_HARASSMENT", "threshold": threshold_options[index]},  # Default to BLOCK_NONE
-            {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": threshold_options[index]},
-            {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": threshold_options[index]},
-            {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": threshold_options[index]},
-        ]
-
             
 #HEADER
 st.markdown('''
@@ -735,7 +699,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', safety_settings=safety_settings, generation_config=generation_config)
     return model
 
 @st.cache_resource
@@ -815,6 +779,24 @@ if len(st.session_state.chat_session) > 0:
 
 #st.session_state.chat.history
                 
+
+cols = st.columns(5)
+
+with cols[0]:
+    image_attachment = st.toggle("Attach image", value=False, help="Activate this mode to attach an image and let the chatbot read it")
+
+with cols[1]:
+    txt_attachment = st.toggle("Attach text file", value=False, help="Activate this mode to attach a text file and let the chatbot read it")
+
+with cols[2]:
+    csv_excel_attachment = st.toggle("Attach CSV or Excel", value=False, help="Activate this mode to attach a CSV or Excel file and let the chatbot read it")
+
+with cols[3]:
+    graphviz_mode = st.toggle("Graphviz mode", value=False, help="Activate this mode to generate a graph with graphviz in .dot from your message")
+
+with cols[4]:
+    pdf_mode = st.toggle("Pdf ask mode", value=False, help="Activate this mode to generate a graph with graphviz in .dot from your message")
+    
 if image_atachment:
     image = st.file_uploader("Upload your image", type=['png', 'jpg', 'jpeg'])
     url = st.text_input("Or paste your image url")
@@ -891,7 +873,7 @@ 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=safety_settings, generation_config=generation_config)
             response.resolve()
         else:
             response = None