Update app.py

app.py

@@ -29,14 +29,14 @@ LANGUAGES = {
 }
 
 def configure_gemini():
-    """Configure Gemini API"""
-    api_key = st.session_state.get('api_key')
-    if not api_key:
-        return False
-    
+    """Configure Gemini API using secrets"""
     try:
+        api_key = st.secrets["GEMINI_API_KEY"]
         genai.configure(api_key=api_key)
         return True
+    except KeyError:
+        st.error("❌ GEMINI_API_KEY not found in secrets. Please add it to your Streamlit secrets.")
+        return False
     except Exception as e:
         st.error(f"Failed to configure Gemini API: {str(e)}")
         return False
@@ -96,28 +96,33 @@ def get_image_description(image: Image.Image, language: str) -> Optional[str]:
     except Exception as e:
         return f"Error describing image: {str(e)}"
 
+def save_to_history(question: str, answer: str, language: str):
+    """Save analysis to history"""
+    if 'analysis_history' not in st.session_state:
+        st.session_state.analysis_history = []
+    
+    st.session_state.analysis_history.append({
+        'question': question,
+        'answer': answer,
+        'language': language
+    })
+
 def main():
     st.title("🖼️ Image Q&A Assistant")
     st.markdown("Upload an image and ask questions about it in multiple languages!")
     
-    # Sidebar for API key and settings
+    # Configure Gemini API from secrets
+    api_configured = configure_gemini()
+    
+    # Sidebar for settings
     with st.sidebar:
         st.header("⚙️ Settings")
         
-        # API Key input
-        api_key = st.text_input(
-            "Gemini API Key",
-            type="password",
-            help="Enter your Google Gemini API key",
-            key="api_key_input"
-        )
-        
-        if api_key:
-            st.session_state.api_key = api_key
-            if configure_gemini():
-                st.success("✅ API key configured successfully!")
-            else:
-                st.error("❌ Invalid API key")
+        # API Status
+        if api_configured:
+            st.success("✅ Gemini API configured successfully!")
+        else:
+            st.error("❌ Please configure GEMINI_API_KEY in secrets")
         
         st.markdown("---")
         
@@ -174,7 +179,7 @@ def main():
     with col2:
         st.subheader("💬 Ask Questions")
         
-        if 'current_image' in st.session_state and st.session_state.api_key:
+        if 'current_image' in st.session_state and api_configured:
             # Question input
             question = st.text_area(
                 "Your question about the image:",
@@ -204,6 +209,9 @@ def main():
                         LANGUAGES[selected_language]
                     )
                     
+                    # Save to history
+                    save_to_history(question, result, selected_language)
+                    
                     st.subheader("🎯 Analysis Result")
                     st.write(result)
             
@@ -214,6 +222,9 @@ def main():
                         LANGUAGES[selected_language]
                     )
                     
+                    # Save to history
+                    save_to_history("General Description", description, selected_language)
+                    
                     st.subheader("📋 Image Description")
                     st.write(description)
             
@@ -223,8 +234,8 @@ def main():
         elif 'current_image' not in st.session_state:
             st.info("📷 Please upload an image first.")
         
-        elif not st.session_state.get('api_key'):
-            st.warning("🔑 Please enter your Gemini API key in the sidebar.")
+        elif not api_configured:
+            st.warning("🔑 Please configure GEMINI_API_KEY in your Streamlit secrets.")
     
     # Results history (optional feature)
     if st.checkbox("📚 Show Analysis History"):