Update app.py

app.py

@@ -61,11 +61,12 @@ if stock_symbol:
     for article in news_articles:
         st.write(f"- {article}")
 
-# Key Financial Entities extraction
+# Key Financial Entities extraction with filtering
 st.subheader("Extracted Key Financial Entities")
 if input_text:
     entities = ner(input_text)
-    for entity in entities:
+    filtered_entities = [entity for entity in entities if entity['score'] > 0.7 and entity['word'].isalpha()]  # Filter low-score and non-alphabetic tokens
+    for entity in filtered_entities:
         st.write(f"Entity: {entity['word']}, Label: {entity.get('entity', 'N/A')}, Score: {entity['score']:.2f}")
 
 # Sentiment Analysis
@@ -75,10 +76,11 @@ if input_text:
     for result in sentiment:
         st.write(f"Sentiment: {result['label']}, Score: {result['score']:.2f}")
 
-# Investment Advice or Strategy Generation
+# Investment Advice or Strategy Generation with better prompt handling
 st.subheader("Investment Advice or Strategy")
 if input_text:
-    advice = text_generator(f"Provide an investment strategy for {selected_example if selected_example != 'Other' else 'the selected stock'}: {input_text}", max_length=50, num_return_sequences=1)
+    prompt = f"Provide a clear and concise investment strategy for {selected_example if selected_example != 'Other' else 'the selected stock'} based on recent news and financial performance. "
+    advice = text_generator(prompt + input_text, max_length=80, num_return_sequences=1)
     st.write(advice[0]['generated_text'])
 else:
     st.write("No investment advice generated. Please select a stock or enter custom text.")