Geek7 commited on
Commit
04f6c13
·
verified ·
1 Parent(s): 52235c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -31
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("AI Stock Trend Predictor")
 
 
 
21
 
22
- # Input for stock symbol
23
- symbol = st.text_input("Enter stock symbol (e.g., AAPL):", "AAPL")
 
24
 
25
  # Fetch yfinance data
26
  stock_data = fetch_yfinance_data(symbol)
27
 
28
- # Calculate indicators
29
- stock_data = calculate_indicators(stock_data)
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()