edit code again

app.py

@@ -24,7 +24,6 @@ def generate_text(prompt, model, max_tokens, temperature):
                 "temperature": temperature,
             },
         )
-        # ✅ Extract and return generated text
         if response and response.candidates:
             return response.candidates[0].content.parts[0].text
         else:
@@ -36,35 +35,30 @@ def generate_text(prompt, model, max_tokens, temperature):
 # ✅ Streamlit Page Configuration
 st.set_page_config(page_title="Gemini AI Text Generator", layout="centered")
 
-# ✅ UI Layout - Title & Description
-st.title("📝 Creative AI Assistant")
-st.markdown("#### Generate AI-powered text with ease. Simply enter a prompt and customize the settings!")
+# ✅ Title & Description
+st.title("🤖 Gemini AI Text Generator")
+st.write("Generate AI-powered text by entering a prompt and customizing the settings.")
 
-# ✅ User Input Fields
-st.markdown("##### ✍🏼 Enter Your Prompt")
-user_input = st.text_area("Write something...", height=150, placeholder="Example: Write a formal article about climate change.")
+# ✅ Input Prompt
+user_input = st.text_area("📝 Enter Your Prompt:", height=150, placeholder="Example: Write a formal article about climate change.")
 
-# ✅ Layout for Options
-col1, col2 = st.columns(2)
+# ✅ Sliders (Aligned Below Prompt)
+temperature = st.slider("🎨 Creativity Level (1-10)", min_value=1, max_value=10, value=5) / 10  
+max_tokens = st.slider("📏 Max Tokens (Word Length)", min_value=50, max_value=1000, value=300, step=50)
 
-with col1:
-    model = st.selectbox("🤖 Choose AI Model", ["gemini-pro", "gemini-pro-code"])
-    max_tokens = st.slider("📏 Max Tokens (Word Length)", min_value=50, max_value=1000, value=300, step=50)
+# ✅ AI Model Selection (Below Sliders)
+model = st.selectbox("🤖 Choose AI Model", ["gemini-pro", "gemini-pro-code"])
 
-with col2:
-    temperature = st.slider("🎨 Creativity Level (1-10)", min_value=1, max_value=10, value=5) / 10  # Scaling to 0.1 - 1.0
-
-# ✅ Generate Button
-st.markdown("### ")
+# ✅ Generate Button (Below Model Selection)
 if st.button("🚀 Generate Text"):
     if user_input.strip():
         with st.spinner("✨ Generating response..."):
             result = generate_text(user_input, model, max_tokens, temperature)
-        st.markdown("### 🎯 Generated Response")
+        st.subheader("🎯 Generated Response")
         st.write(result)
     else:
         st.warning("⚠️ Please enter a prompt before generating text.")
 
 # ✅ Footer
 st.markdown("---")
-st.caption("🔹 *Powered by Gemini AI* | 🔹 *Developed with Streamlit*")
+st.write("🔹 *Powered by Gemini AI* | 🔹 *Developed with Streamlit*")