Update app.py

app.py

@@ -7,10 +7,17 @@ def get_stock_info(symbol):
         price = stock_info.get_live_price(symbol)
         # Fetch company information
         company_info = stock_info.get_quote_table(symbol)
-        return f"Company: {company_info['Name']}\nSector: {company_info['Sector']}\nMarket Cap: {company_info['Market Cap']}\nStock Price: ${price:.2f}"
+        
+        # Handling 'None' values gracefully if they exist
+        company_name = company_info.get('Name', 'N/A')
+        sector = company_info.get('Sector', 'N/A')
+        market_cap = company_info.get('Market Cap', 'N/A')
+        
+        return f"Company: {company_name}\nSector: {sector}\nMarket Cap: {market_cap}\nStock Price: ${price:.2f}"
     except Exception as e:
         return f"Error retrieving data: {e}"
 
+# Gradio Interface
 iface = gr.Interface(fn=get_stock_info,
                      inputs=gr.Textbox(label="Enter Stock Symbol (e.g., AAPL)"),
                      outputs="text",