|
import gradio as gr |
|
import yfinance as yf |
|
import pandas as pd |
|
from sklearn.linear_model import LinearRegression |
|
import plotly.graph_objects as go |
|
|
|
markdown_content = """ |
|
# Asset Price Prediction Tool |
|
|
|
## Introduction |
|
This tool uses historical stock price data to predict future prices. It's designed to provide insights into potential price trends based on past performance. |
|
|
|
## How to Use |
|
1. **Enter the Ticker Symbol:** Input the stock ticker (e.g., 'AAPL' for Apple Inc.). |
|
2. **Select Start and End Dates:** Choose the historical data range for analysis. Dates must be entered in the format YYYY-MM-DD (e.g., 2023-01-01). |
|
3. **Set Prediction Days:** Decide how many days into the future you want to predict. |
|
4. **Submit:** Click 'Run' to view the predictions. |
|
|
|
## How It Works |
|
- **Data Fetching:** The tool fetches historical closing prices of the specified asset using `yfinance` for the date range you provide. |
|
- **Model Training:** It then trains a linear regression model on this data. The model learns the relationship between dates and closing prices during this period. |
|
- **Making Predictions:** Based on the learned relationship, the model attempts to predict future prices for the number of days you specified. |
|
|
|
## Understanding Linear Regression |
|
- Linear regression is a statistical method that models the relationship between a dependent variable and one or more independent variables. |
|
- In this tool, the dependent variable is the asset's price, and the independent variable is time (dates). |
|
- The model assumes a linear relationship (a straight line trend) between dates and prices. |
|
- It's important to note that this method works best when the relationship between the data points is linear and may not capture complex market dynamics. |
|
|
|
## Interpreting Data |
|
- **Historical Prices:** Displayed as a solid blue line, representing actual past closing prices. |
|
- **Predicted Prices:** Shown as a solid red line, indicating the model's predictions. |
|
- **Limitations:** The predictions are based on historical trends and do not account for unforeseen market events or changes in market conditions. They should be used as a guideline rather than definitive financial advice. |
|
|
|
Remember, investing in the stock market involves risks, and past performance is not indicative of future results. |
|
""" |
|
|
|
def train_predict_wrapper(ticker, start_date, end_date, prediction_days): |
|
|
|
data = yf.download(ticker, start=start_date, end=end_date) |
|
data = data["Close"] |
|
|
|
|
|
data.index = (data.index - pd.Timestamp("1970-01-01")) // pd.Timedelta('1s') |
|
|
|
|
|
X = data.index.values[:-prediction_days].reshape(-1, 1) |
|
y = data.values[:-prediction_days] |
|
model = LinearRegression() |
|
model.fit(X, y) |
|
|
|
|
|
last_timestamp = data.index[-1] |
|
future_timestamps = pd.date_range(start=pd.to_datetime(last_timestamp, unit='s'), |
|
periods=prediction_days, freq='D') |
|
future_timestamps = (future_timestamps - pd.Timestamp("1970-01-01")) // pd.Timedelta('1s') |
|
X_future = future_timestamps.values.reshape(-1, 1) |
|
|
|
|
|
predicted_prices = model.predict(X_future) |
|
|
|
|
|
historical_prices = go.Scatter( |
|
x=pd.to_datetime(data.index, unit='s'), |
|
y=data.values, |
|
mode="lines", |
|
name="Historical Prices" |
|
) |
|
predicted_prices_trace = go.Scatter( |
|
x=pd.to_datetime(future_timestamps, unit='s'), |
|
y=predicted_prices, |
|
mode="lines", |
|
name="Predicted Prices" |
|
) |
|
|
|
|
|
fig = go.Figure() |
|
fig.add_trace(historical_prices) |
|
fig.add_trace(predicted_prices_trace) |
|
fig.update_layout( |
|
title="Asset Price Prediction", |
|
xaxis_title="Date", |
|
yaxis_title="Price", |
|
legend_title_text="Data" |
|
) |
|
|
|
return fig |
|
|
|
|
|
interface = gr.Interface( |
|
fn=train_predict_wrapper, |
|
inputs=[ |
|
gr.Textbox(label="Ticker Symbol"), |
|
gr.Textbox(label="Start Date (YYYY-MM-DD)"), |
|
gr.Textbox(label="End Date (YYYY-MM-DD)"), |
|
gr.Slider(minimum=1, maximum=365, step=1, label="Prediction Days"), |
|
], |
|
outputs="plot", |
|
description=markdown_content |
|
) |
|
|
|
|
|
interface.launch() |