final changes to poetry app

app.py

@@ -5,14 +5,14 @@ from dotenv import load_dotenv
 
 # Load environment variables
 load_dotenv()
-api_key = os.getenv("GEMINI_API_KEY")  # Use Google Gemini API key
+api_key = os.getenv("GEMINI_API_KEY")  # Gets the Google Gemini API key
 
 # Check if API key is set
 if not api_key:
     st.error("❌ Error: Google API key is missing. Please check your .env file.")
     st.stop()
 
-# Initialize Google Gemini AI client
+# Set up Google Gemini AI
 genai.configure(api_key=api_key)
 
 # Streamlit UI
@@ -20,41 +20,45 @@ st.set_page_config(page_title="REYREY's AI Poetry Translator", page_icon="🎭",
 st.title("📝 REYREY's Poetry Translator 🎭")
 st.write("Enter a phrase, sentence, or paragraph, and I'll transform it into poetry!")
 
-# User input
+# Get user input
 text_input = st.text_area("Enter text:", placeholder="Type something here...")
 
-# Poetry style selection
+# Let user choose a poetry style
 poetry_style = st.selectbox(
     "Choose a poetry style:",
     ["Shakespearean", "Haiku", "Free Verse", "Sonnet", "Limerick", "Rhyming Couplets"]
 )
 
-# Function to call Google Gemini API
-def generate_poetry(text, style):
+# Generates a poem using Google Gemini AI
+def generate_poetry(text, style, retry=False):
     try:
         prompt = f"Convert the following text into {style} poetry:\n\n{text}\n\nPoem:"
-        
-        model = genai.GenerativeModel("gemini-pro")  # Use Google's Gemini Pro model
+        model = genai.GenerativeModel("gemini-pro")  # Uses Google's Gemini Pro model
         response = model.generate_content(prompt)
 
-        # ✅ Check if response has valid candidates
-        if response.candidates:
-            for candidate in response.candidates:
-                if candidate.finish_reason == "STOP":
-                    return candidate.content.parts[0].text.strip()
+        # Checks if AI returned a valid response
+        if response and hasattr(response, "text") and response.text:
+            return response.text.strip()
 
-        # ✅ Handle safety filter blocking content
-        return "❌ AI Response was blocked due to content safety filters."
+        # Retries with a softer prompt if the first attempt is blocked
+        if not retry:
+            softer_prompt = f"Write a short {style} poem based on this idea: {text}"
+            response = model.generate_content(softer_prompt)
+            if response and hasattr(response, "text") and response.text:
+                return response.text.strip()
+
+        # Returns a friendly message if AI refuses to generate a poem
+        return "⚠️ AI couldn't generate a poem. Try rewording your input."
 
     except Exception as e:
         return f"❌ Error generating poem: {str(e)}"
 
-# Generate poetry on button click
+# Runs when the user clicks the button
 if st.button("Generate Poetry ✨"):
     if text_input:
         with st.spinner("Creating your poem... 🎭"):
             poem = generate_poetry(text_input, poetry_style)
-            if poem.startswith("❌ Error") or poem.startswith("❌ AI Response"):
+            if poem.startswith("❌ Error"):
                 st.error(poem)
             else:
                 st.success("Here’s your poetic masterpiece:")