Update app.py

app.py

@@ -1,32 +1,26 @@
 import requests
 from bs4 import BeautifulSoup
+from transformers import pipeline
+import gradio as gr
 
-def get_stock_data(symbol):
-    url = f"https://finance.yahoo.com/quote/{symbol}"
+def get_web_page(url):
     response = requests.get(url)
     soup = BeautifulSoup(response.text, 'html.parser')
+    return soup.text
 
-    data = {}
-    data['symbol'] = symbol
-    data['price'] = soup.find_all('div', {'class': 'My(6px) Pos(r) smartphone_Mt(6px)'})[0].find('span').text
-    data['change'] = soup.find_all('div', {'class': 'My(6px) Pos(r) smartphone_Mt(6px)'})[0].find_all('span')[1].text
-
-    return data
+def answer_question(question, context):
+    qa_pipeline = pipeline("question-answering", model="tiiuae/falcon-40b")
+    result = qa_pipeline({
+        'context': context,
+        'question': question
+    })
+    return result['answer']
 
-def get_news(symbol):
-    url = f"https://finance.yahoo.com/quote/{symbol}/news?p={symbol}"
-    response = requests.get(url)
-    soup = BeautifulSoup(response.text, 'html.parser')
-
-    news = []
-    for item in soup.find_all('h3', {'class': 'Mb(5px)'}):
-        news_item = {}
-        news_item['title'] = item.text
-        news_item['link'] = item.find('a')['href']
-        news.append(news_item)
-
-    return news
+def app(symbol, question):
+    url = f"https://finance.yahoo.com/quote/{symbol}"
+    context = get_web_page(url)
+    answer = answer_question(question, context)
+    return answer
 
-symbol = input("Please enter a stock symbol: ")
-print(get_stock_data(symbol))
-print(get_news(symbol))
+iface = gr.Interface(fn=app, inputs=["text", "text"], outputs="text")
+iface.launch()