Update app.py

app.py

@@ -9,56 +9,62 @@ st.set_page_config(page_title="Pattern Completion", page_icon="🤖")
 if 'history' not in st.session_state:
     st.session_state.history = [
         "input: dogs",
-        "output: cats",
+        "output: cats", 
         "input: apple",
         "output: samsung",
         "input: b",
         "output: c",
-        "input: friend",
+        "input: friend", 
         "output: enemy",
         "input: oil",
         "output: water",
     ]
 
-# Configure Gemini
-genai.configure(api_key=st.secrets["GEMINI_API_KEY"])
+# Create the Streamlit interface
+st.title("Pattern Completion")
 
-# Create the model
-generation_config = {
-    "temperature": 1,
-    "top_p": 0.95,
-    "top_k": 64,
-    "max_output_tokens": 8192,
-    "response_mime_type": "text/plain",
-}
+# API key input
+api_key = st.text_input("Enter your Gemini API key:", type="password")
 
-model = genai.GenerativeModel(
-    model_name="gemini-exp-1206",
-    generation_config=generation_config,
-)
+if api_key:
+    # Configure Gemini
+    genai.configure(api_key=api_key)
 
-# Create the Streamlit interface
-st.title("Pattern Completion")
+    # Create the model
+    generation_config = {
+        "temperature": 1,
+        "top_p": 0.95,
+        "top_k": 64,
+        "max_output_tokens": 8192,
+        "response_mime_type": "text/plain",
+    }
 
-# Input text box
-user_input = st.text_input("Enter your input:")
+    model = genai.GenerativeModel(
+        model_name="gemini-exp-1206",
+        generation_config=generation_config,
+    )
 
-if st.button("Generate"):
-    if user_input:
-        # Create prompt with history and new input
-        prompt = st.session_state.history.copy()
-        prompt.extend([f"input: {user_input}", "output: "])
-        
-        # Generate response
-        try:
-            response = model.generate_content(prompt)
-            output = response.text.strip()
-            
-            # Update history
-            st.session_state.history.extend([f"input: {user_input}", f"output: {output}"])
-            
-            # Display result
-            st.success(f"Output: {output}")
+    # Input text box
+    user_input = st.text_input("Enter your input:")
+
+    if st.button("Generate"):
+        if user_input:
+            # Create prompt with history and new input
+            prompt = st.session_state.history.copy()
+            prompt.extend([f"input: {user_input}", "output: "])
             
-        except Exception as e:
-            st.error(f"An error occurred: {str(e)}")
+            # Generate response
+            try:
+                response = model.generate_content(prompt)
+                output = response.text.strip()
+                
+                # Update history
+                st.session_state.history.extend([f"input: {user_input}", f"output: {output}"])
+                
+                # Display result
+                st.success(f"Output: {output}")
+                
+            except Exception as e:
+                st.error(f"An error occurred: {str(e)}")
+else:
+    st.warning("Please enter your Gemini API key to use the app")