Update app.py

app.py

@@ -1,42 +1,56 @@
 import streamlit as st
-import datetime
+import os
+import sys
+
+# Add the 'utils' directory to sys.path so Python can find modules within it
+sys.path.append(os.path.join(os.path.dirname(__file__), "utils"))
+
+# Importing the necessary function from model_inference.py
 from model_inference import generate_forex_signals
-from config import RISK_LEVELS
+from config import RISK_LEVELS  # Assuming config.py is in the root directory
+
+# Streamlit page configuration
+st.set_page_config(page_title="AI Forex Bot", page_icon=":guardsman:", layout="wide")
+
+# Title of the application
+st.title("AI Forex Trading Bot")
+
+# Description of how the bot works
+st.markdown("""
+    This AI-powered Forex Bot provides real-time trading signals using a combination of historical data, sentiment analysis, and macroeconomic factors.
+    You can enter your trading capital and select a risk level to get personalized trading recommendations.
+""")
 
-# Streamlit app setup
-st.title("AI Forex Bot")
+# Sidebar for user inputs
 st.sidebar.header("User Input")
 
-# Input: Trading Capital
-trading_capital = st.sidebar.number_input(
-    "Trading Capital (USD)", 
-    min_value=1.0, 
-    max_value=10000.0, 
-    step=1.0
-)
-
-# Input: Risk Level
-risk_level = st.sidebar.selectbox("Market Risk Level", options=list(RISK_LEVELS.keys()))
-
-# Input: User's current time
-user_timezone_time = st.sidebar.time_input(
-    "Enter Current Time (Your Time Zone)", 
-    value=datetime.datetime.now().time()
-)
-
-# Generate signals button
-if st.sidebar.button("Generate Signals"):
-    # Call backend logic
-    forex_signals = generate_forex_signals(trading_capital, risk_level)
-
-    # Adjust entry and exit times based on user's time
-    entry_time = datetime.datetime.combine(datetime.date.today(), user_timezone_time)
-    exit_time = entry_time + datetime.timedelta(hours=2)
-
-    # Display results
-    st.write("### Results")
-    st.write(f"**Profitable Currency Pair**: {forex_signals['currency_pair']}")
-    st.write(f"**Entry Time**: {entry_time.strftime('%Y-%m-%d %H:%M:%S')}")
-    st.write(f"**Exit Time**: {exit_time.strftime('%Y-%m-%d %H:%M:%S')}")
-    st.write(f"**ROI**: {forex_signals['roi']}%")
-    st.write(f"**Signal Strength**: {forex_signals['signal_strength']}")
+# Get user trading capital (1 USD to 10,000 USD)
+trading_capital = st.sidebar.number_input("Enter Trading Capital (USD)", min_value=1, max_value=10000, value=1000)
+
+# Get user market risk level (Low, Medium, High)
+market_risk = st.sidebar.selectbox("Select Market Risk Level", options=["Low", "Medium", "High"])
+
+# Get user's preferred timezone (this can be used to adjust entry/exit time)
+user_timezone = st.sidebar.text_input("Enter Your Timezone (e.g., UTC, EST, PST)", "UTC")
+
+# Display the inputs
+st.write(f"**Trading Capital**: ${trading_capital}")
+st.write(f"**Market Risk Level**: {market_risk}")
+st.write(f"**Timezone**: {user_timezone}")
+
+# When user presses the "Get Trading Signals" button, run the model inference
+if st.sidebar.button("Get Trading Signals"):
+    # Call the function to generate forex signals from model_inference.py
+    try:
+        signals = generate_forex_signals(trading_capital, market_risk, user_timezone)
+        
+        # Display the results in the main area
+        st.subheader("Recommended Forex Trade")
+        st.write(f"**Currency Pair**: {signals['currency_pair']}")
+        st.write(f"**Entry Time**: {signals['entry_time']}")
+        st.write(f"**Exit Time**: {signals['exit_time']}")
+        st.write(f"**ROI**: {signals['roi']}%")
+        st.write(f"**Signal Strength**: {signals['signal_strength']}")
+
+    except Exception as e:
+        st.error(f"Error: {str(e)}")