Update app.py

app.py

@@ -11,6 +11,10 @@ def to_markdown(text):
 # Function to generate content using Gemini API
 def generate_gemini_content(prompt, model_name='gemini-pro', image=None):
     model = genai.GenerativeModel(model_name)
+    if model_name == 'gemini-pro-vision' and not image:
+        st.warning("Please add an image to use the gemini-pro-vision model.")
+        return None
+
     if image:
         response = model.generate_content([prompt, image])
     else:
@@ -42,21 +46,23 @@ def main():
     # Generate content on button click
     if st.button("Generate Content"):
         st.markdown("### Generated Content:")
-        if image_file:
-            # If an image is provided, use gemini-pro-vision model
-            image = Image.open(image_file)
-            response = generate_gemini_content(prompt, model_name='gemini-pro-vision', image=image)
+        if model_name == 'gemini-pro-vision':
+            if not image_file:
+                st.warning("Please provide an image for the gemini-pro-vision model.")
+            else:
+                image = Image.open(image_file)
+                response = generate_gemini_content(prompt, model_name=model_name, image=image)
         else:
             response = generate_gemini_content(prompt, model_name=model_name)
 
-        # Display the generated content in Markdown format
-        if response.candidates:
-            # Check if candidates are available
-            parts = response.candidates[0].content.parts
-            generated_text = parts[0].text if parts else "No content generated."
-            st.markdown(to_markdown(generated_text))
-        else:
-            st.warning("No candidates found in the response.")
+        # Display the generated content in Markdown format if response is available
+        if response:
+            if response.candidates:
+                parts = response.candidates[0].content.parts
+                generated_text = parts[0].text if parts else "No content generated."
+                st.markdown(to_markdown(generated_text))
+            else:
+                st.warning("No candidates found in the response.")
 
 if __name__ == "__main__":
     main()