Update app.py

app.py

@@ -31,22 +31,12 @@ def main():
 
     if user_text:
         # Available Models
-        # Sentiment Analysis
-        sentiment_checkbox = st.checkbox("Sentiment Analysis")
-
-        # Emotion Analysis
-        emotion_checkbox = st.checkbox("Emotion Analysis")
-
-        # Named Entity Recognition
-        ner_checkbox = st.checkbox("Named Entity Recognition")
-
-        # Toxicity Analysis
-        toxicity_checkbox = st.checkbox("Toxicity Analysis")
-
+        analysis_type = st.selectbox("Select Analysis Type", ["", "Sentiment Analysis", "Emotion Analysis", "Named Entity Recognition", "Toxicity Analysis"])
+        
         # Run custom and display outputs
-        st.header("Function Outputs:")
+        st.header("Function Output:")
 
-        if sentiment_checkbox:
+        if analysis_type == "Sentiment Analysis":
             st.subheader("Sentiment Analysis:")
             
             # Parse JSON data
@@ -60,7 +50,7 @@ def main():
                     st.write(f"Label: {label}, Score: {score}")
 
 
-        if emotion_checkbox:
+        if analysis_type == "Emotion Analysis":
             st.subheader("Emotion Analysis:")
             
             # Parse JSON data
@@ -73,7 +63,7 @@ def main():
                     score = entry["score"]
                     st.write(f"Label: {label}, Score: {score}")
 
-        if ner_checkbox:
+        if analysis_type == "Named Entity Recognition":
             st.subheader("Named Entity Recognition:")
             
             # Parse JSON data
@@ -81,14 +71,15 @@ def main():
 
             # Extract and display data
             for entry in data:
-                entity_group = entry["entity_group"]
-                score = entry["score"]
-                word = entry["word"]
-                start = entry["start"]
-                end = entry["end"]
-                st.write(f"Word: {word}, Entity Group: {entity_group}, Score: {score}, Start: {start}, End: {end}")
-
-        if toxicity_checkbox:
+                for entry in entry:
+                    entity_group = entry["entity_group"]
+                    score = entry["score"]
+                    word = entry["word"]
+                    start = entry["start"]
+                    end = entry["end"]
+                    st.write(f"Word: {word}, Entity Group: {entity_group}, Score: {score}, Start: {start}, End: {end}")
+
+        if analysis_type == "Toxicity Analysis":
             st.subheader("Toxicity Analysis:")
             
             # Parse JSON data
@@ -103,3 +94,5 @@ def main():
 
 if __name__ == "__main__":
     main()
+
+