Add link and fix google finance

app.py

@@ -4,15 +4,11 @@ import requests
 from bs4 import BeautifulSoup
 
 example_tickers = [
-    {"Ticker": "PLD", "Company Name": "Prologis Inc"},
-    {"Ticker": "PSA", "Company Name": "Public Storage"},
-    {"Ticker": "O", "Company Name": "Realty Income Corp"},
-    {
-        "Ticker": "META",
-        "Company Name": "Meta Platforms",
-    },
-    {"Ticker": "AMZN", "Company Name": "Amazon.com"},
-    {"Ticker": "MSFT", "Company Name": "Microsoft Corporation"},
+    {"Ticker": "PLD", "Company Name": "Prologis Inc", "Market": "NYSE"},
+    {"Ticker": "PSA", "Company Name": "Public Storage", "Market": "NYSE"},
+    {"Ticker": "META", "Company Name": "Meta Platforms", "Market": "NASDAQ"},
+    {"Ticker": "MSFT", "Company Name": "Microsoft Corporation", "Market": "NASDAQ"},
+    {"Ticker": "AMZN", "Company Name": "Amazon.com", "Market": "NASDAQ"},
 ]
 
 
@@ -41,21 +37,32 @@ def get_esg_from_yahoo_finance(row):
         row["Environmental Risk Score"] = None
         row["Social Risk Score"] = None
         row["Governance Risk Score"] = None
+
+    row["Yahoo Finance Link"] = (
+        f"[Yahoo Link](https://finance.yahoo.com/quote/"
+        + row.Ticker
+        + "/sustainability)"
+    )
     return row
 
 
 def get_esg_score_from_google_finance(row):
+    print(row)
     headers = {
         "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/103.0.5060.134 Safari/537.36"
     }
+    link = f"https://www.google.com/finance/quote/{row.Ticker}:{row.Market}"
+
+    html = requests.get(link, headers=headers, timeout=30).text
 
-    html = requests.get(
-        f"https://www.google.com/finance/quote/META:NASDAQ", headers=headers, timeout=30
-    ).text
     soup = BeautifulSoup(html, "html.parser")
     scores = soup.find_all("div", {"class": "IPIeJ"})
+    if scores:
+        row["CDP Score"] = scores[0].find("div", {"class": "P6K39c"}).text
 
-    row["CDP Score"] = scores[0].find("div", {"class": "P6K39c"}).text
+    else:
+        row["CDP Score"] = None
+    row["Google Finance Link"] = f"[Google Finance Link]({link})"
 
     return row
 
@@ -65,26 +72,46 @@ example_input_data = pd.DataFrame(example_tickers)
 
 inputs = [
     gr.Dataframe(
-        row_count=(6, "dynamic"),
-        col_count=(1, "dynamic"),
+        row_count=(5, "dynamic"),
+        col_count=(3, "dynamic"),
         label="Input Data",
         interactive=1,
+        headers=[
+            "Ticker",
+            "Company Name",
+            "Market",
+        ],
     )
 ]
 
 outputs = [
     gr.Dataframe(
-        row_count=(6, "dynamic"),
-        col_count=(7, "fixed"),
+        row_count=(5, "dynamic"),
+        col_count=(10, "fixed"),
         label="ESG Scores",
         headers=[
             "Ticker",
             "Company Name",
+            "Market",
             "Total ESG Risk Score",
             "Environmental Risk Score",
             "Social Risk Score",
             "Governance Risk Score",
+            "Yahoo Finance Link",
             "CDP Score",
+            "Google Finance Link",
+        ],
+        datatype=[
+            "str",
+            "str",
+            "str",
+            "str",
+            "str",
+            "str",
+            "str",
+            "markdown",
+            "str",
+            "markdown",
         ],
     )
 ]