Update app.py
Browse files
app.py
CHANGED
@@ -1,48 +1,27 @@
|
|
1 |
import streamlit as st
|
2 |
import yfinance as yf
|
3 |
import pandas as pd
|
4 |
-
from Pandas_Market_Predictor import Pandas_Market_Predictor
|
5 |
|
6 |
def fetch_yfinance_data(symbol):
|
7 |
data = yf.download(symbol, start="2022-01-01", end="2022-12-31", interval="5m")
|
8 |
return data
|
9 |
|
10 |
-
def calculate_indicators(data):
|
11 |
-
# Convert all columns to numeric
|
12 |
-
data = data.apply(pd.to_numeric, errors='coerce')
|
13 |
-
|
14 |
-
# Example: Simple condition for doji and inside
|
15 |
-
data['Doji'] = abs(data['Close'] - data['Open']) <= 0.01 * (data['High'] - data['Low'])
|
16 |
-
data['Inside'] = (data['High'] < data['High'].shift(1)) & (data['Low'] > data['Low'].shift(1))
|
17 |
-
return data
|
18 |
-
|
19 |
def main():
|
20 |
-
st.title("
|
|
|
|
|
|
|
21 |
|
22 |
-
|
23 |
-
|
|
|
24 |
|
25 |
# Fetch yfinance data
|
26 |
stock_data = fetch_yfinance_data(symbol)
|
27 |
|
28 |
-
#
|
29 |
-
|
30 |
-
|
31 |
-
# Create predictor
|
32 |
-
my_market_predictor = Pandas_Market_Predictor(stock_data)
|
33 |
-
|
34 |
-
# Predict Trend
|
35 |
-
indicators = ["Doji", "Inside"]
|
36 |
-
trend = my_market_predictor.Trend_Detection(indicators, 10)
|
37 |
-
|
38 |
-
# Display results
|
39 |
-
st.subheader("Predicted Trend:")
|
40 |
-
st.write("Buy Trend :", trend['BUY'])
|
41 |
-
st.write("Sell Trend :", trend['SELL'])
|
42 |
-
st.write("Hold Trend :", trend['HOLD'])
|
43 |
-
|
44 |
-
# Delete the DataFrame to release memory
|
45 |
-
del stock_data
|
46 |
|
47 |
if __name__ == "__main__":
|
48 |
main()
|
|
|
1 |
import streamlit as st
|
2 |
import yfinance as yf
|
3 |
import pandas as pd
|
|
|
4 |
|
5 |
def fetch_yfinance_data(symbol):
|
6 |
data = yf.download(symbol, start="2022-01-01", end="2022-12-31", interval="5m")
|
7 |
return data
|
8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
9 |
def main():
|
10 |
+
st.title("Stock Trend Predictor")
|
11 |
+
|
12 |
+
# User input for stock symbol
|
13 |
+
symbol = st.text_input("Enter Stock Symbol (e.g., IBM):")
|
14 |
|
15 |
+
if not symbol:
|
16 |
+
st.warning("Please enter a valid stock symbol.")
|
17 |
+
st.stop()
|
18 |
|
19 |
# Fetch yfinance data
|
20 |
stock_data = fetch_yfinance_data(symbol)
|
21 |
|
22 |
+
# Display the raw data
|
23 |
+
st.subheader("Raw Data:")
|
24 |
+
st.write(stock_data)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
|
26 |
if __name__ == "__main__":
|
27 |
main()
|