Update app.py

app.py

@@ -11,6 +11,7 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Function to get response from Gemini model
 def get_gemini_response(input_prompt, image_data, genre, length, language, mood):
     model = genai.GenerativeModel('gemini-1.5-flash')
+    
     # If both image and text are provided
     if image_data and input_prompt:
         full_prompt = f"""
@@ -73,23 +74,34 @@ st.markdown("### Upload your image or provide a prompt to create a Vision around
 col1, col2 = st.columns([2, 3])  # 2 + 3 = 5 parts total
 
 with col1:
-    input_prompt = st.text_input("Input Prompt (optional):", placeholder="Enter your prompt here...")
+    # File uploader for image upload
     uploaded_file = st.file_uploader("Choose an image...", type=["jpg", "jpeg", "png"])
-    
-    genre = st.selectbox("Select Genre:", ["story", "shayari", "sher", "poem", "quote"])
-    length = st.selectbox("Select Length:", ["short", "long"])
-    language = st.selectbox("Select Language:", ["English", "Spanish"])
-    mood = st.selectbox("Select Mood:", ["emotional", "sad", "happy", "horror", "comedy", "romantic"])
 
-    # Generate button
-    submit = st.button("Generate")
+    # Expander for additional options, closed by default
+    with st.expander("Additional Options", expanded=False):  # Changed to expanded=False
+        input_prompt = st.text_input("Input Prompt (optional):", placeholder="Enter your prompt here...")
+        genre = st.selectbox("Select Genre:", ["story", "shayari", "sher", "poem", "quote"])
+        length = st.selectbox("Select Length:", ["short", "long"])
+        language = st.selectbox("Select Language:", ["English", "Spanish"])
+        mood = st.selectbox("Select Mood:", ["emotional", "sad", "happy", "horror", "comedy", "romantic"])
 
+    # Generate button with color
+    submit = st.button("Generate", key="generate_button", style="color: white; background-color: green; width: 100%;")
+
+# Display uploaded image
 if uploaded_file is not None:
     image = Image.open(uploaded_file)
     col1.image(image, caption="Uploaded Image", use_column_width=True)
 
 # Handle submit action
 if submit:
+    # Use data from input fields whether or not the expander is open
+    input_prompt = st.session_state.get('input_prompt', None)
+    genre = st.session_state.get('genre', None)
+    length = st.session_state.get('length', None)
+    language = st.session_state.get('language', None)
+    mood = st.session_state.get('mood', None)
+    
     if uploaded_file or input_prompt:
         try:
             image_data = input_image_setup(uploaded_file)