Update app.py

app.py

@@ -7,23 +7,95 @@ from agno.agent import Agent
 from agno.models.groq import Groq
 from agno.tools.yfinance import YFinanceTools
 from agno.tools.duckduckgo import DuckDuckGoTools
+import base64
 import groq
+from functools import wraps
 from dotenv import load_dotenv
 import time
+from finance_agent import FinanceAgent
+from pathlib import Path
+# UI enhancement
+def add_bg_from_local(image_path):
+    with open(image_path, "rb") as image_file:
+        encoded_string = base64.b64encode(image_file.read()).decode()
+    
+    return f"""
+    <style>
+    .stApp {{
+        background-image: url("data:image/png;base64,{encoded_string}");
+        background-size: cover;
+        background-position: center;
+        background-repeat: no-repeat;
+        background-attachment: fixed;
+    }}
+    </style>
+    """
+
+def apply_custom_style():
+    # Add the background image
+    image_path = "/Users/aditya/Desktop/finance_agent/23115991-fffc-4f67-a116-581a1a9046c8.png"
+    st.markdown(add_bg_from_local(image_path), unsafe_allow_html=True)
+    
+    # Add the rest of the custom styling
+    st.markdown("""
+    <style>
+        .main .block-container {
+            padding-top: 2rem;
+            max-width: 1000px;
+            background-color: rgba(255, 255, 255, 0.85);
+            border-radius: 10px;
+            padding: 20px;
+        }
+        h1 {
+            color: #2E4057;
+            text-align: center;
+            margin-bottom: 2rem;
+            padding-bottom: 1rem;
+            border-bottom: 2px solid #E3B448;
+        }
+        .stButton > button {
+            border-radius: 10px;
+            background-color: #2E4057;
+            color: white;
+            font-weight: 600;
+            width: 100%;
+        }
+        .stTextInput > div > div > input {
+            border-radius: 10px;
+        }
+        .stExpander {
+            background-color: #f8f9fa;
+            border-radius: 10px;
+            padding: 0.75rem;
+            margin-bottom: 1rem;
+        }
+        .stChart {
+            background-color: #f8f9fa;
+            border-radius: 10px;
+            padding: 1rem;
+            box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+        }
+        div[data-testid="stVerticalBlock"] div[style*="flex-direction: column;"] {
+            gap: 1rem;
+        }
+    </style>
+    """, unsafe_allow_html=True)
+apply_custom_style()
+
+
+
+# Apply the custom styling
+apply_custom_style()
 
 # Load environment variables first
 load_dotenv()
 
-# Check if API key exists and set it manually if needed
-if not os.getenv("GROQ_API_KEY"):
-    st.error("GROQ_API_KEY not found in environment variables. Please add it to your .env file.")
-    st.stop()
 
 # Set API key explicitly
 groq.api_key = os.getenv("GROQ_API_KEY")
 
 # Import agent modules after setting API key
-from agents.finance_agent import FinanceAgent
+
 
 # Initialize only the finance agent first
 finance_agent = FinanceAgent()