Update app.py

app.py

@@ -39,9 +39,6 @@ for message in chat_history:
 
 user_input = st.text_input("") 
 
-# File uploader for images
-uploaded_file = st.file_uploader("Upload an image (optional)")
-
 if user_input: 
   chat_history.append({"role": role, "parts": [{"text": user_input}]})
 
@@ -85,7 +82,10 @@ st.title("Image Description Generator")
 
 uploaded_file = st.file_uploader("Upload an image here", type=["png", "jpg", "jpeg"])
 
-if uploaded_file:
+# Text input for asking questions about the image
+image_question = st.text_input("Ask something about the image:")
+
+if uploaded_file and image_question:
    image_parts = [
      {
        "mime_type": uploaded_file.type,
@@ -94,7 +94,7 @@ if uploaded_file:
    ]
 
    prompt_parts = [
-     "Describe the image:",
+     image_question,
      image_parts[0],
    ]
 
@@ -105,4 +105,4 @@ if uploaded_file:
    )
    
    response = model.generate_content(prompt_parts)
-   st.markdown(f"**Model's description:** {response.text}")
+   st.markdown(f"**Model's answer:** {response.text}")