Upload 17 files

app.py

@@ -265,5 +265,5 @@ with tab2:
 
     else:
         st.write(
-            "No specific document chosen. Please mention Ticker and Duration in the Question."
+            "No specific document/documents found. Please mention Ticker and Duration in the Question."
         )

utils/entity_extraction.py

@@ -1,4 +1,3 @@
-import re
 from ast import literal_eval
 from nltk.stem import PorterStemmer, WordNetLemmatizer
 
@@ -9,14 +8,18 @@ def generate_ner_docs_prompt(query):
     prompt = """USER: Extract the company names and time duration mentioned in the question. The entities should be extracted in the following format: {"companies": list of companies mentioned in the question,"start-duration": ("start-quarter", "start-year"), "end-duration": ("end-quarter", "end-year")}. Return {"companies": None, "start-duration": (None, None), "end-duration": (None, None)} if the entities are not found.
 
 Examples:
-What is Intel's update on the server chip roadmap and strategy for Q1 2019?
-{"companies": ["Intel"], "start-duration": ("Q1", "2019"), "end-duration": ("Q1", "2019")}
+What is Nvidia's visibility in the data center business in Q2 2020?
+{"companies": ["Nvidia"], "start-duration": ("Q2", "2020"), "end-duration": ("Q2", "2020")}
+What is Intel's update on the server chip road map and strategy for 2019?
+{"companies": ["Intel"], "start-duration": ("Q1", "2019"), "end-duration": ("Q4", "2019")}
 What are the opportunities and challenges in the Indian market for Amazon in 2016?
 {"companies": ["Amazon"], "start-duration": ("Q1", "2016"), "end-duration": ("Q4", "2016")}
 What did analysts ask about the Cisco's Webex?
 {"companies": ["Cisco"], "start-duration": (None, None), "end-duration": (None, None)}
-What is the comparative performance analysis between Intel and AMD in key overlapping segments such as PC, Gaming, and Data Centers in Q2 to Q3 2018?
-{"companies": ["Intel", "AMD"], "start-duration": ("Q2", "2018"), "end-duration": ("Q3", "2018")}
+What is the comparative performance analysis between Intel and AMD in key overlapping segments such as PC, Gaming, and Data Centers in Q1 to Q3 2016?
+{"companies": ["Intel", "AMD"], "start-duration": ("Q1", "2016"), "end-duration": ("Q3", "2016")}
+How has the growth been for AMD in the PC market in 2020?
+{"companies": ["AMD"], "start-duration": ("Q1", "2020"), "end-duration": ("Q4", "2020")}
 How did Microsoft and Amazon perform in terms of reliability and scalability of cloud for the years 2016 and 2017?
 {"companies": ["Microsoft", "Amazon"], "start-duration": ("Q1", "2016"), "end-duration": ("Q4", "2017")}"""
     input_prompt = f"""###Input: {query}