Update app.py

app.py

@@ -144,26 +144,25 @@ if user_input:
         max_output_tokens=max_output_tokens,
         # add other settings if needed
     )
-
-    # Generate model response
-    try:
-        if model_name == "gemini-pro":
-            model = genai.GenerativeModel('gemini-pro')
-            response = model.generate_content(
-                contents=[user_input],
-                generation_config=generation_config
-            )
-        elif model_name == "gemini-pro-vision":
-            images = [Image.open(file).convert('RGB') for file in uploaded_files]
-            image_prompts = [{'mime_type': 'image/png', 'data': image.tobytes()} for image in images]
-            model = genai.GenerativeModel('gemini-pro-vision')
-            response = model.generate_content(
-                contents=[user_input] + image_prompts,
-                generation_config=generation_config
-            )
-    except Exception as e:
-        st.write(f"An error occurred: {e}")
-        return
+# Generate model response
+try:
+   if model_name == "gemini-pro":
+       model = genai.GenerativeModel('gemini-pro')
+       response = model.generate_content(
+           contents=[user_input],
+           generation_config=generation_config
+       )
+   elif model_name == "gemini-pro-vision":
+       images = [Image.open(file).convert('RGB') for file in uploaded_files]
+       image_prompts = [{'mime_type': 'image/png', 'data': image.tobytes()} for image in images]
+       model = genai.GenerativeModel('gemini-pro-vision')
+       response = model.generate_content(
+           contents=[user_input] + image_prompts,
+           generation_config=generation_config
+       )
+except Exception as e:
+   st.write(f"An error occurred: {e}")
+   # No need to return here
 
     # Add model response to chat history
     st.session_state["chat_history"].append({"role": "model", "parts": [{"text": response}]})