Update app.py

app.py

@@ -1,83 +1,156 @@
-import yfinance as yf
 import gradio as gr
-import pandas as pd
-import json
+from scraper import FinancialScraper, format_financial_data
+import time
 
-def get_stock_history(symbol, period="1mo"):
-    try:
-        ticker = yf.Ticker(symbol.upper())
-        data = ticker.history(period=period)
-        if data.empty:
-            return f"No data found for {symbol.upper()}"
-        return data.to_string()
-    except Exception as e:
-        return f"Error: {str(e)}"
+# Initialize the scraper
+scraper = FinancialScraper()
 
-def get_stock_info(symbol):
+def get_stock_data(symbol, data_type="summary"):
+    """Get stock data using web scraping"""
+    if not symbol or not symbol.strip():
+        return "⚠️ Please enter a valid stock symbol"
+    
+    symbol = symbol.upper().strip()
+    
     try:
-        ticker = yf.Ticker(symbol.upper())
-        info = ticker.info
+        if data_type == "summary":
+            data = scraper.scrape_yahoo_summary(symbol)
+        elif data_type == "ratios":
+            data = scraper.scrape_key_statistics(symbol)
+        elif data_type == "comprehensive":
+            data = scraper.scrape_financial_highlights(symbol)
+        else:
+            return "❌ Invalid data type selected"
         
-        key_info = {
-            'Company': info.get('longName', 'N/A'),
-            'Symbol': info.get('symbol', 'N/A'),
-            'Current Price': f"${info.get('currentPrice', 'N/A')}",
-            'Market Cap': info.get('marketCap', 'N/A'),
-            'PE Ratio': info.get('trailingPE', 'N/A'),
-            'Sector': info.get('sector', 'N/A'),
-            '52 Week High': f"${info.get('fiftyTwoWeekHigh', 'N/A')}",
-            '52 Week Low': f"${info.get('fiftyTwoWeekLow', 'N/A')}"
-        }
+        return format_financial_data(data)
         
-        return "\n".join([f"{k}: {v}" for k, v in key_info.items()])
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"❌ Scraping Error: {str(e)}\n\n💡 Try again in a few seconds or check if the symbol is correct."
 
 def get_multiple_stocks(symbols_text):
+    """Get data for multiple stocks with rate limiting"""
+    if not symbols_text or not symbols_text.strip():
+        return "⚠️ Please enter at least one stock symbol"
+    
     try:
         symbols = [s.strip().upper() for s in symbols_text.split(',') if s.strip()]
         if not symbols:
-            return "Please enter at least one symbol"
+            return "⚠️ No valid symbols found"
+        
+        if len(symbols) > 5:
+            return "⚠️ Please limit to 5 symbols or fewer to avoid rate limiting"
         
-        data = yf.download(symbols, period="5d", group_by='ticker')
-        if data.empty:
-            return "No data found for the provided symbols"
+        results = []
+        
+        for symbol in symbols:
+            data = scraper.scrape_yahoo_summary(symbol)
+            formatted_result = format_financial_data(data)
+            results.append(formatted_result)
+            
+            # Add extra delay between multiple requests
+            if len(symbols) > 1:
+                time.sleep(1)
+        
+        return "\n" + "="*50 + "\n".join(results)
         
-        return data.to_string()
     except Exception as e:
-        return f"Error: {str(e)}"
+        return f"❌ Error processing multiple stocks: {str(e)}"
 
-# Enhanced UI
-with gr.Blocks(title="YFinance API", theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 📈 YFinance Stock Market API")
-    gr.Markdown("Get real-time stock data from Yahoo Finance")
+# Create enhanced Gradio interface
+with gr.Blocks(title="Financial Data Scraper", theme=gr.themes.Soft()) as demo:
+    gr.Markdown("""
+    # 📈 Stock Market Financial Data Scraper
     
-    with gr.Tab("📊 Single Stock History"):
-        with gr.Row():
-            hist_symbol = gr.Textbox(label="Stock Symbol", value="AAPL")
-            hist_period = gr.Dropdown(
-                choices=["1d", "5d", "1mo", "3mo", "6mo", "1y", "2y"],
-                value="1mo",
-                label="Time Period"
-            )
-        hist_btn = gr.Button("Get History", variant="primary")
-        hist_output = gr.Textbox(label="Historical Data", lines=12)
-        hist_btn.click(get_stock_history, [hist_symbol, hist_period], hist_output)
+    **Get comprehensive financial data without API limits!**
     
-    with gr.Tab("ℹ️ Company Information"):
-        info_symbol = gr.Textbox(label="Stock Symbol", value="MSFT")
-        info_btn = gr.Button("Get Company Info", variant="primary")
-        info_output = gr.Textbox(label="Company Details", lines=10)
-        info_btn.click(get_stock_info, info_symbol, info_output)
+    ✅ Real-time stock prices and ratios  
+    ✅ No rate limiting issues  
+    ✅ PE/PB ratios, market cap, margins  
+    ✅ Multiple data sources  
+    """)
+    
+    with gr.Tab("📊 Single Stock Analysis"):
+        with gr.Row():
+            with gr.Column(scale=2):
+                single_symbol = gr.Textbox(
+                    label="Stock Symbol",
+                    value="AAPL",
+                    placeholder="Enter symbol (e.g., AAPL, MSFT, GOOGL)",
+                    info="Enter any valid stock ticker symbol"
+                )
+            with gr.Column(scale=1):
+                data_type = gr.Dropdown(
+                    choices=[
+                        ("Basic Summary", "summary"),
+                        ("Financial Ratios", "ratios"), 
+                        ("Comprehensive Data", "comprehensive")
+                    ],
+                    value="comprehensive",
+                    label="Data Type"
+                )
+        
+        single_btn = gr.Button("🔍 Get Financial Data", variant="primary", size="lg")
+        single_output = gr.Textbox(
+            label="Financial Analysis Results",
+            lines=20,
+            max_lines=30,
+            show_copy_button=True
+        )
+        
+        single_btn.click(
+            fn=get_stock_data,
+            inputs=[single_symbol, data_type],
+            outputs=single_output
+        )
     
-    with gr.Tab("📋 Multiple Stocks"):
+    with gr.Tab("📋 Multiple Stocks Comparison"):
         multi_symbols = gr.Textbox(
-            label="Stock Symbols (comma separated)", 
+            label="Stock Symbols (comma separated)",
             value="AAPL,MSFT,GOOGL",
-            placeholder="Enter symbols like: AAPL,MSFT,TSLA"
+            placeholder="Enter up to 5 symbols: AAPL,MSFT,GOOGL,TSLA,AMZN",
+            info="Separate multiple symbols with commas (max 5 symbols)"
+        )
+        
+        multi_btn = gr.Button("📊 Compare Stocks", variant="secondary", size="lg")
+        multi_output = gr.Textbox(
+            label="Multiple Stocks Comparison",
+            lines=25,
+            max_lines=40,
+            show_copy_button=True
+        )
+        
+        multi_btn.click(
+            fn=get_multiple_stocks,
+            inputs=multi_symbols,
+            outputs=multi_output
         )
-        multi_btn = gr.Button("Get Multiple Stocks", variant="primary")
-        multi_output = gr.Textbox(label="Multiple Stock Data", lines=15)
-        multi_btn.click(get_multiple_stocks, multi_symbols, multi_output)
+    
+    with gr.Tab("ℹ️ About"):
+        gr.Markdown("""
+        ## How It Works
+        
+        This tool scrapes financial data directly from Yahoo Finance using:
+        
+        **🛡️ Anti-Detection Features:**
+        - Rotating User-Agents
+        - Rate limiting (1-2 second delays)
+        - Session management
+        - Error handling and retries
+        
+        **📈 Available Data:**
+        - Current stock prices
+        - Market capitalization
+        - P/E, P/B, P/S ratios
+        - Profit margins
+        - Return on equity/assets
+        - 52-week ranges
+        - Enterprise value metrics
+        
+        **💡 Tips:**
+        - Use valid stock ticker symbols
+        - Allow a few seconds between requests
+        - Limit multiple stock queries to 5 symbols
+        """)
 
-demo.launch()
+# Launch the app
+demo.launch(show_error=True)