Upload app.py

app.py

@@ -25,6 +25,8 @@ load_dotenv()
 import datetime
 import fitz  # PyMuPDF
     
+weathe_key = os.getenv("WEATHER_KEY")
+
 history=[
   {
     "parts": [
@@ -519,7 +521,6 @@ def user_input(user_question):
     
 url = "https://y39t47-8080.csb.app/search"
 
-weathe_key = os.getenv("WEATHER_KEY")
 def print_weather_data(api_key, location):
     base_url = "http://api.weatherstack.com/forecast"
     params = {"access_key": api_key, "query": location}
@@ -576,25 +577,45 @@ st.set_page_config(
 )
 # Path: Main.py
 #Author: Sergio Demis Lopez Martinez
+generation_config = {
+    "temperature": 0.9,
+    "top_p": 1,
+    "top_k": 1,
+    "max_output_tokens": 2048,
+}
 #------------------------------------------------------------
 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"
-  }
+    {"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"},
 ]
+
+with st.sidebar:
+    st.subheader("Settings")
+    temperature = st.slider("Temperature", 0.1, 1.0, generation_config["temperature"])
+    top_p = st.slider("Top P", 0.1, 1.0, generation_config["top_p"])
+    top_k = st.slider("Top K", 1, 100, generation_config["top_k"])
+    max_output_tokens = st.slider("Max Output Tokens", 100, 5000, generation_config["max_output_tokens"])
+    apply_button = st.button("Apply")
+
+# Check if the "Apply" button is clicked
+if apply_button:
+    # Update generation_config based on user input
+    generation_config["temperature"] = temperature
+    generation_config["top_p"] = top_p
+    generation_config["top_k"] = top_k
+    generation_config["max_output_tokens"] = max_output_tokens
+    
+with st.sidebar:
+    st.subheader("Safety Settings")
+    for setting in SAFETY_SETTINGS:
+        category = setting["category"]
+        threshold = st.selectbox(f"{category} Threshold", ["BLOCK_NONE", "BLOCK_LOW_AND_ABOVE", "BLOCK_MID_AND_ABOVE", "BLOCK_HIGH"], index=0)
+
+        # Update safety settings based on user input
+        setting["threshold"] = threshold
+        
 #HEADER
 st.markdown('''
 Powered by Hk4crprasad <img src="https://i.ibb.co/BVJ2tCG/OIG-47-modified.png" width="20" height="20">
@@ -651,7 +672,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)
+    model = genai.GenerativeModel('gemini-pro', safety_settings=SAFETY_SETTINGS, generation_config=generation_config)
     return model
 
 @st.cache_resource
@@ -836,7 +857,7 @@ if prompt:
     
     with st.spinner(spinertxt):
         if len(prmt['parts']) > 1:
-            response = vision.generate_content(prmt['parts'],stream=True,safety_settings=SAFETY_SETTINGS)
+            response = vision.generate_content(prmt['parts'],stream=True,safety_settings=SAFETY_SETTINGS, generation_config=generation_config)
             response.resolve()
         else:
             response = None