Update app.py

app.py

@@ -1,10 +1,9 @@
 # Importing required modules
-import pandas as pd
+import pandas as pd # for manipulating financial statements in dataframes
 import numpy as np
-import numpy as np
-import plotly.express as px
+import plotly.express as px # for visualizing results in interactive plots
 
-# To extract and parse fundamental data like beta and growth estimates from finviz website
+# To extract and parse fundamental data like beta and growth estimates from finviz website's HTML
 import requests
 from bs4 import BeautifulSoup as bs
 
@@ -12,16 +11,18 @@ from bs4 import BeautifulSoup as bs
 from urllib.request import urlopen
 import json
 
+def get_jsonparsed_data(url):
+    response = urlopen(url)
+    data = response.read().decode("utf-8")
+    return json.loads(data)
+
 # For Gradio App
 import gradio as gr
 
-import os
-# uncomment and set API Key in the environment variable below
-# or you can choose to set it using any other method you know
-#os.environ['FMP_API_KEY'] = "your_api_key"
 
-# read the environment variable to use in API requests later
-apiKey = os.environ['FMP_API_KEY']
+# To read the environment variable to use in API requests later
+import os
+apiKey = os.environ['FMP_API_KEY'] # the environment variable is set in HuggingFace Spaces directly
 
 
 ############################################################################################################
@@ -31,11 +32,6 @@ apiKey = os.environ['FMP_API_KEY']
 # Financialmodelingprep api url
 base_url = "https://financialmodelingprep.com/api/v3/"
 
-def get_jsonparsed_data(url):
-    response = urlopen(url)
-    data = response.read().decode("utf-8")
-    return json.loads(data)
-
 # get financial statements using financial modelling prep API
 def get_financial_statements(ticker):
     # quarterly cash flow statements for calculating latest trailing twelve months (TTM) free cash flow
@@ -55,7 +51,7 @@ def get_financial_statements(ticker):
     
     # quarterly balance sheet statements
     columns_drop = ['acceptedDate', 'calendarYear', 'period', 'symbol', 'reportedCurrency', 'cik', 'fillingDate', 'link', 'finalLink']
-    q_balance_statement = pd.DataFrame(get_jsonparsed_data(base_url+'balance-sheet-statement/' + ticker + '?' + '&apikey=' + apiKey))
+    q_balance_statement = pd.DataFrame(get_jsonparsed_data(base_url+'balance-sheet-statement/' + ticker + '?period=quarter' + '&apikey=' + apiKey))
     q_balance_statement = q_balance_statement.set_index('date').drop(columns_drop, axis=1)
     q_balance_statement = q_balance_statement.apply(pd.to_numeric, errors='coerce')