Update app.py

app.py

@@ -50,8 +50,13 @@ def main():
             response = generate_gemini_content(prompt, model_name=model_name)
 
         # Display the generated content in Markdown format
-        generated_text = response.candidates[0].content.parts[0].text
-        st.markdown(to_markdown(generated_text))
+        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.")
 
 if __name__ == "__main__":
     main()