Update app.py

app.py

@@ -23,24 +23,18 @@ def analyze_financial_news():
 
     if analyze_clicked:
         # Perform text classification on the input text
-        results = classification(text)
-
-        # Filter only the news related to "Energy | Oil"
-        energy_oil_news = [news for news in results if news["label"] == "Energy | Oil"]
-        
-        st.write("Energy | Oil News:", energy_oil_news)  # Debug statement to check the Energy | Oil news list
-        
-        if energy_oil_news:
-            # If there are news related to "Energy | Oil", perform sentiment analysis for each news
-            for idx, news in enumerate(energy_oil_news, start=1):
-                # Print the structure of each news to find the correct key for the news text
-                st.write(f"News {idx} structure:")
-                st.write(news)
-                
-                # Uncomment the following line once you identify the correct key for the news text
-                # sentiment_results = sentiment_analysis(news["<correct_key_for_text>"])[0]
+        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"])
         else:
-            st.write("No news relevant to Energy | Oil.")
+            st.write("The provided news is not relevant to Energy | Oil.")
 
 def main():
     analyze_financial_news()