Update app.py

app.py

@@ -1,29 +1,44 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load the text classification model
-text_classifier = pipeline("text-classification", model="distilbert-base-uncased")
+# Load the sentiment analysis model
+sentiment_analyzer = pipeline("sentiment-analysis", model="distilbert-base-uncased")
 
 def main():
-    st.title("How Good is my Writing?")
+    st.title("Feedback Analyzer for Creative Writing")
     
-    # Text input area for the user
-    text = st.text_area("Enter your text:")
+    # Text input area for the user's creative writing
+    writing = st.text_area("Enter your creative writing:")
 
-    if st.button("Rate it"):
-        if text:
-            analyze_writing_quality(text)
+    if st.button("Analyze Writing"):
+        if writing:
+            analyze_writing_feedback(writing)
         else:
-            st.warning("Please enter some text.")
+            st.warning("Please enter your creative writing.")
 
-def analyze_writing_quality(text):
-    classification_result = text_classifier(text)
-    predicted_label = classification_result[0]['label']
-    predicted_score = classification_result[0]['score']
+def analyze_writing_feedback(writing):
+    sentiment_result = sentiment_analyzer(writing)
+    sentiment_label = sentiment_result[0]['label']
+    sentiment_score = sentiment_result[0]['score']
     
-    st.subheader("Writing Quality Analysis Result:")
-    st.write(f"Predicted Quality: {predicted_label}")
-    st.write(f"Confidence Score: {predicted_score:.2f}")
+    feedback = generate_feedback(sentiment_label)
+    
+    st.subheader("Feedback for Creative Writing:")
+    st.write("Sentiment Analysis Result:")
+    st.write(f"Sentiment: {sentiment_label}")
+    st.write(f"Confidence Score: {sentiment_score:.2f}")
+    st.write("Feedback:")
+    st.write(feedback)
+
+def generate_feedback(sentiment_label):
+    if sentiment_label == "LABEL_1":
+        feedback = "Your writing has a positive sentiment! It evokes a sense of optimism and positivity."
+    elif sentiment_label == "LABEL_0":
+        feedback = "Your writing has a negative sentiment. It might be helpful to focus on brighter and more uplifting themes."
+    else:
+        feedback = "Your writing is neutral. Consider adding more emotional depth to enhance the impact."
+    
+    return feedback
 
 if __name__ == "__main__":
     main()