Update app.py

app.py

@@ -13,14 +13,23 @@ st.markdown("""
         background-color: #0e1117;
         color: white;
     }
-    .positive {background: linear-gradient(90deg, #3fc1c9, #4caf50); border-radius: 10px; padding: 1rem;}
-    .negative {background: linear-gradient(90deg, #f05454, #b33939); border-radius: 10px; padding: 1rem;}
-    .neutral {background: linear-gradient(90deg, #607d8b, #455a64); border-radius: 10px; padding: 1rem;}
+    .positive {background: linear-gradient(90deg, #3fc1c9, #4caf50); border-radius: 10px; padding: 1rem; margin-bottom:1rem;}
+    .negative {background: linear-gradient(90deg, #f05454, #b33939); border-radius: 10px; padding: 1rem; margin-bottom:1rem;}
+    .neutral {background: linear-gradient(90deg, #607d8b, #455a64); border-radius: 10px; padding: 1rem; margin-bottom:1rem;}
+    .essay-box {
+        background-color: #1e1e2e;
+        border: 1px solid #333;
+        border-radius: 10px;
+        padding: 1.5rem;
+        margin-top: 1rem;
+        color: #f1f1f1;
+        box-shadow: 0 0 10px rgba(0,0,0,0.3);
+    }
     </style>
 """, unsafe_allow_html=True)
 
 st.title("🧠 AI Sentiment Essay Generator")
-st.caption("Analyze the emotion of your topic and let AI craft an expressive essay aligned with it 🎭")
+st.caption("Analyze your topic’s sentiment and let AI write an expressive essay aligned with it 🎭")
 
 # ----------------------------------
 # Model Initialization (cached)
@@ -76,7 +85,7 @@ if generate_btn and user_prompt.strip():
     # Generate Essay
     # ----------------------------------
     with st.spinner("Generating essay with AI..."):
-        gen_prompt = f"Write a {sentiment_label.lower()} and expressive essay about: {user_prompt}"
+        gen_prompt = f"Write a {sentiment_label.lower()} and expressive essay with a short title about: {user_prompt}"
 
         input_ids = tokenizer(gen_prompt, return_tensors="pt").input_ids
         output_ids = text_model.generate(
@@ -89,8 +98,16 @@ if generate_btn and user_prompt.strip():
         )
         generated_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
 
+        # Clean up output: remove initial instruction phrases
+        cleaned_text = generated_text.replace(gen_prompt, "").strip()
+        cleaned_text = cleaned_text.replace("Title:", "**Title:**").strip()
+
+    # ----------------------------------
+    # Display Essay in Styled Box
+    # ----------------------------------
     st.markdown("### 📝 AI-Generated Essay")
-    st.write(generated_text)
+
+    st.markdown(f"<div class='essay-box'>{cleaned_text}</div>", unsafe_allow_html=True)
 
     st.markdown("---")
     st.caption("✨ Powered by RoBERTa + Qwen | Streamlit Frontend")