Update app.py

app.py

@@ -178,6 +178,33 @@ if 'history' not in st.session_state:
 # Create the Streamlit interface
 st.title("Pattern Completion")
 
+with st.sidebar:
+    st.header("Model Configuration")
+    
+    # Model selection dropdown
+    model_name = st.selectbox(
+        "Select Model",
+        ["gemini-exp-1121", "gemini-exp-1206", "gemini-1.5-pro"]
+    )
+    
+    # Temperature slider
+    temperature = st.slider(
+        "Temperature",
+        min_value=0.0,
+        max_value=1.0,
+        value=1.0,
+        step=0.1
+    )
+    
+    # Top_p slider
+    top_p = st.slider(
+        "Top P",
+        min_value=0.0,
+        max_value=1.0,
+        value=0.95,
+        step=0.05
+    )
+
 # API key input
 api_key = st.text_input("Enter your Gemini API key:", type="password")
 
@@ -187,15 +214,15 @@ if api_key:
 
     # Create the model
     generation_config = {
-        "temperature": 1,
-        "top_p": 0.95,
+        "temperature": temperature,
+        "top_p": top_p,
         "top_k": 64,
         "max_output_tokens": 8192,
         "response_mime_type": "text/plain",
     }
 
     model = genai.GenerativeModel(
-        model_name="gemini-exp-1206",
+        model_name=model_name,
         generation_config=generation_config,
     )
 
@@ -207,7 +234,7 @@ if api_key:
             # Create prompt with history and new input
             prompt = st.session_state.history.copy()
             prompt.extend([f"input: {user_input}", "output: "])
-            
+            print(prompt)
             # Generate response
             try:
                 response = model.generate_content(prompt)