Spaces:
Sleeping
Sleeping
File size: 3,700 Bytes
b74f860 ff6a3ca b74f860 e5f21ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 |
import gradio as gr
import yfinance as yf
import plotly.graph_objects as go
import plotly.express as px
from datetime import datetime
import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
# Function to construct the pie chart
def get_pie_graph(data):
data['Change'] = data['Open'] - data['Close'].shift(1)
data['Movement'] = data['Change'].apply(lambda x : 'Up' if x > 0 else 'Down')
fig, ax = plt.subplots()
ax.pie(data['Movement'].value_counts(), labels=[f"{label}: {count}" for label, count in zip(data['Movement'].unique(), data['Movement'].value_counts().tolist())])
return fig
# Function to construct the chart with markers
def get_trading_markers(data, t, tll):
val = data['Close']
th = []
marker = []
for i in range(len(data)):
if(i == 0):
th.append(0)
else:
x = val[i] - val[i-1]
x = (x / val[i-1] )*10
th.append(abs(x))
data['Value'] = th
data['Marker'] = np.where(data['Value'] > t, 'yes', 'no')
df = data[data['Marker'] == 'yes']
df = df.Close
x = pd.DataFrame({tll: data["Close"]})
plt.plot(x)
plt.scatter(df.index, df, marker = 'x', c = 'b' , s = 60, zorder=2)
plt.title(tll)
return plt
def get_stock_data(start_date, end_date, stock_ticker, graph_type, t):
# Validate date format
try:
start_date = datetime.strptime(start_date, "%Y-%m-%d").date()
end_date = datetime.strptime(end_date, "%Y-%m-%d").date()
except ValueError:
return "Invalid date format. Please use the YYYY-MM-DD format."
# Fetch stock data using Yahoo Finance API
data = yf.download(stock_ticker, start=start_date, end=end_date)
data.reset_index(inplace=True) # Reset index to get separate "Date" column
# Return a different graph type depending on which option the user selected
match graph_type:
case 'Line chart of open prices':
# Create the line plot using Plotly Express
line_fig = px.line(data, x='Date', y='Open', title='Open Price')
return line_fig
case 'Candle chart of stocks':
candle_fig = go.Figure(data=[go.Candlestick(x=data['Date'],
open=data['Open'],
high=data['High'],
low=data['Low'],
close=data['Close'])])
return candle_fig
case 'Chart of volume traded':
bar_fig = px.line(data, x = 'Date', y = 'Volume', title = 'Volume Traded')
return bar_fig
case 'Pie chart of stock movement':
pie_fig = get_pie_graph(data)
return pie_fig
case 'Chart of trading markers':
trade_markers_fig = get_trading_markers(data,t, stock_ticker)
return trade_markers_fig
outputs = [gr.Plot(label="Plot")]
iface = gr.Interface(
fn=get_stock_data,
inputs=[
gr.inputs.Textbox(placeholder="YYYY-MM-DD", label="Start Date"),
gr.inputs.Textbox(placeholder="YYYY-MM-DD", label="End Date"),
gr.inputs.Textbox(placeholder="AAPL", label="Stock Ticker"),
# Dropdown of different choices of graphs to display
gr.inputs.Dropdown(choices=['Line chart of open prices',
'Candle chart of stocks',
'Chart of volume traded',
'Pie chart of stock movement',
'Chart of trading markers'],
label='Graph type'),
gr.inputs.Number(label="Marker value")
],
outputs=outputs,
title="Stock Data Viewer",
description="Enter the start date, end date, and stock ticker to view the stock data.",
)
iface.launch()
|