Update app.py

app.py

@@ -53,13 +53,13 @@ def is_telugu_text(text):
     cleaned_text = re.sub(r'[\u0C00-\u0C7F0-9\s\.\,\!\?]', '', text)
     return len(cleaned_text) == 0
 
-# Session state setup
+# Session state
 if "text_input" not in st.session_state:
     st.session_state.text_input = ""
-if "result_shown" not in st.session_state:
-    st.session_state.result_shown = False
+if "result" not in st.session_state:
+    st.session_state.result = None
 
-# Examples
+# Example inputs
 st.subheader("Try one of the following examples:")
 examples = [
     "ఈ ఆహారం చాలా చెడుగా ఉంది",
@@ -67,7 +67,8 @@ examples = [
     "నేను ఈ వార్తలకు చాలా బాధపడ్డాను",
     "ఈ చిత్రం నాకు చాలా భయంకరంగా ఉంది",
     "ఈ సెల్ఫీ చాలా అందంగా ఉంది",
-    "ఈ వాతావరణం నాకు చాలా ఉష్ణంగా ఉంది"
+    "ఈ వాతావరణం నాకు చాలా ఉష్ణంగా ఉంది",
+    "ఈ సినిమా కి 5 స్టార్ ఇచ్చాను"
 ]
 
 for i in range(0, len(examples), 2):
@@ -77,26 +78,30 @@ for i in range(0, len(examples), 2):
             example = examples[i + j]
             if cols[j].button(example[:30] + "..."):
                 st.session_state.text_input = example
-                st.session_state.result_shown = False
+                if not is_telugu_text(example):
+                    st.session_state.result = "error"
+                else:
+                    st.session_state.result = classifier(example)[0]
 
-# Text Area for user input
-input_text = st.text_area("Enter text to analyze sentiment:", height=150)
+# Input text area
+input_text = st.text_area("Enter text to analyze sentiment:", value=st.session_state.text_input, height=150)
 
 # Analyze button
 if st.button("Analyze Sentiment"):
+    st.session_state.text_input = input_text
     if not input_text.strip():
         st.warning("Please enter some text to analyze!")
-        st.session_state.result_shown = False
+        st.session_state.result = None
     elif not is_telugu_text(input_text):
-        st.error("Please enter valid **Telugu** text only.")
-        st.session_state.result_shown = False
+        st.session_state.result = "error"
     else:
-        st.session_state.text_input = input_text
-        st.session_state.result_shown = True
+        st.session_state.result = classifier(input_text)[0]
 
-# Output
-if st.session_state.result_shown:
-    result = classifier(st.session_state.text_input)
-    raw_label = result[0]['label']
-    sentiment, emoji = label_map.get(raw_label, (raw_label, ""))
-    st.markdown(f'<div class="radium-label">Sentiment: {sentiment} {emoji}</div>', unsafe_allow_html=True) 
+# Display result
+if st.session_state.result:
+    if st.session_state.result == "error":
+        st.error("Please enter valid **Telugu** text only (digits allowed).")
+    else:
+        label = st.session_state.result['label']
+        sentiment, emoji = label_map.get(label, (label, ""))
+        st.markdown(f'<div class="radium-label">Sentiment: {sentiment} {emoji}</div>', unsafe_allow_html=True)