Update app.py

app.py

@@ -9,17 +9,13 @@ def to_markdown(text):
     return textwrap.indent(text, '> ', predicate=lambda _: True)
 
 # Function to generate content using Gemini API
-def generate_gemini_content(prompt, model_name='gemini-pro', image=None):
+def generate_gemini_content(prompt, model_name='gemini-pro-vision', image=None):
     model = genai.GenerativeModel(model_name)
-    if model_name == 'gemini-pro-vision' and not image:
+    if 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:
-        response = model.generate_content(prompt)
-
+    response = model.generate_content([prompt, image])
     return response
 
 # Streamlit app
@@ -30,8 +26,8 @@ def main():
     api_key = st.text_input("Enter your Gemini API key:")
     genai.configure(api_key=api_key)
 
-    # Choose a model
-    model_name = st.selectbox("Select a Gemini model", ["gemini-pro", "gemini-pro-vision"])
+    # Set the model to 'gemini-pro-vision'
+    model_name = 'gemini-pro-vision'
 
     # Get user input prompt
     prompt = st.text_area("Enter your prompt:")
@@ -46,23 +42,20 @@ def main():
     # Generate content on button click
     if st.button("Generate Content"):
         st.markdown("### Generated Content:")
-        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)
+        if not image_file:
+            st.warning("Please provide an image for the gemini-pro-vision model.")
         else:
-            response = generate_gemini_content(prompt, model_name=model_name)
-
-        # 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.")
+            image = Image.open(image_file)
+            response = generate_gemini_content(prompt, model_name=model_name, image=image)
+
+            # 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()