Update app.py

app.py

@@ -23,18 +23,24 @@ def analyze_financial_news():
 
     if analyze_clicked:
         # Perform text classification on the input text
-        results = classification(text)[0]
-
-        # Check if the classification is "Energy | Oil"
-        if results["label"] == "Energy | Oil":
-            # If the news is related to Energy | Oil, perform sentiment analysis
-            sentiment_results = sentiment_analysis(text)[0]
-            
-            # Display the sentiment analysis result
-            st.write("Sentiment:", sentiment_results["label"])
-            st.write("Sentiment Score:", sentiment_results["score"])
+        results = classification(text)
+
+        # Filter only the news related to "Energy | Oil"
+        energy_oil_news = [news for news in results if news["label"] == "Energy | Oil"]
+        
+        if energy_oil_news:
+            # If there are news related to "Energy | Oil", perform sentiment analysis for each news
+            for news in energy_oil_news:
+                text = news["sequence"]
+                sentiment_results = sentiment_analysis(text)[0]
+
+                # Display the sentiment analysis result for each news
+                st.write("Original Text:", text)
+                st.write("Sentiment:", sentiment_results["label"])
+                st.write("Sentiment Score:", sentiment_results["score"])
+                st.write("---")
         else:
-            st.write("The provided news is not relevant to Energy | Oil.")
+            st.write("No news relevant to Energy | Oil.")
 
 def main():
     analyze_financial_news()