Create app.py

app.py

@@ -0,0 +1,292 @@
+import streamlit as st
+import requests
+import json
+import os
+import hashlib
+from datetime import datetime
+from dotenv import load_dotenv
+from config import MODELS
+load_dotenv(".env", override=True)
+
+if 'authenticated' not in st.session_state:
+    st.session_state.authenticated = False
+if 'username' not in st.session_state:
+    st.session_state.username = None
+
+FIXED_USERNAME = os.getenv("USERNAME")
+FIXED_PASSWORD_HASH = hashlib.sha256(os.getenv("PASSWORD").encode()).hexdigest()
+
+
+st.set_page_config(
+    page_title="AI Health Coach",
+    page_icon="💪",
+    layout="wide",
+)
+
+st.markdown("""
+<style>
+    .main-header {
+        font-size: 2.5rem;
+        color: #1E88E5;
+        text-align: center;
+        margin-bottom: 1rem;
+    }
+    .workout-card {
+        background-color: #f0f2f6;
+        border-radius: 10px;
+        padding: 20px;
+        margin-bottom: 15px;
+    }
+    .day-header {
+        color: #1E88E5;
+        font-size: 1.3rem;
+        font-weight: bold;
+    }
+    .exercise-name {
+        font-weight: bold;
+        color: #333;
+    }
+    .exercise-detail {
+        color: #555;
+        margin-left: 10px;
+    }
+    .section-header {
+        font-size: 1.5rem;
+        color: #333;
+        margin-top: 1rem;
+        margin-bottom: 0.5rem;
+    }
+    .login-container {
+        max-width: 500px;
+        margin: 0 auto;
+        padding: 2rem;
+        background-color: #f9f9f9;
+        border-radius: 10px;
+        box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
+    }
+    .login-header {
+        text-align: center;
+        margin-bottom: 1.5rem;
+    }
+    .logout-btn {
+        position: absolute;
+        top: 10px;
+        right: 10px;
+    }
+</style>
+""", unsafe_allow_html=True)
+
+
+def hash_password(password):
+    return hashlib.sha256(password.encode()).hexdigest()
+
+def verify_login(username, password):
+    hashed_password = hash_password(password)
+    return username == FIXED_USERNAME and hashed_password == FIXED_PASSWORD_HASH
+
+
+def show_login_page():
+    st.markdown("<h1 class='main-header'>AI Zach Bouery</h1>", unsafe_allow_html=True)
+    
+    with st.container():
+        st.markdown("<div class='login-container'>", unsafe_allow_html=True)
+        
+        st.markdown("<h2 class='login-header'>Login</h2>", unsafe_allow_html=True)
+        
+        username = st.text_input("Username")
+        password = st.text_input("Password", type="password")
+        
+        if st.button("Login", use_container_width=True):
+            if verify_login(username, password):
+                st.session_state.authenticated = True
+                st.session_state.username = username
+                st.success("Login successful!")
+                st.experimental_rerun()
+            else:
+                st.error("Invalid username or password")
+        
+        st.markdown("</div>", unsafe_allow_html=True)
+        
+def show_main_app():
+    st.markdown("<h1 class='main-header'>AI Zach Bouery </h1>", unsafe_allow_html=True)
+    
+    col1, col2, col3 = st.columns([1, 1, 1])
+    with col3:
+        st.markdown("<div style='text-align: right;'>", unsafe_allow_html=True)
+        if st.button("Logout"):
+            st.session_state.authenticated = False
+            st.session_state.username = None
+            st.experimental_rerun()
+        st.markdown(f"Logged in as: **{st.session_state.username}**")
+        st.markdown("</div>", unsafe_allow_html=True)
+    
+    if 'openrouter_config' not in st.session_state:
+        st.session_state.openrouter_config = {"api_key": os.getenv('OPENROUTER_API_KEY')}
+    if 'workout_plan' not in st.session_state:
+        st.session_state.workout_plan = None
+    if 'generation_time' not in st.session_state:
+        st.session_state.generation_time = None
+    if 'model_used' not in st.session_state:
+        st.session_state.model_used = None
+    
+    with st.sidebar:
+        st.header("Configuration")
+
+        openrouter_api_key = os.getenv("OpenrouterAPIKey")
+
+        st.session_state.openrouter_config["api_key"] = openrouter_api_key
+        
+        st.subheader("Model Selection")
+        model_options = list(MODELS.keys())
+        selected_model = st.selectbox("Choose an LLM", model_options)
+        
+        st.subheader("User Information")
+        user_age = st.number_input("Age", min_value=18, max_value=100, value=30)
+        user_weight = st.number_input("Weight (kg)", min_value=30, max_value=250, value=70)
+        user_height = st.number_input("Height (cm)", min_value=100, max_value=250, value=170)
+        
+        fitness_level = st.selectbox(
+            "Fitness Level",
+            options=["Beginner", "Intermediate", "Advanced"]
+        )
+        health_conditions = st.multiselect(
+            "Health Conditions (if any)",
+            ["None", "High Blood Pressure", "Diabetes", "Asthma", "Joint Pain", "Back Pain", "Heart Condition", "Other"]
+        )
+        
+        time_available = st.text_input("Time Available Per Day (minutes)")
+    
+    col1, col2 = st.columns([1, 1])
+
+    with col1:
+        st.markdown("<div class='section-header'>Workout Goals</div>", unsafe_allow_html=True)
+        primary_goal = st.selectbox(
+            "Primary Goal",
+            ["Weight Loss", "Muscle Building", "Cardiovascular Health", "Flexibility & Mobility", "General Fitness"]
+        )
+        secondary_goals = st.multiselect(
+            "Secondary Goals (Optional)",
+            ["Stress Reduction", "Better Sleep", "Increased Energy", "Improved Posture", "Core Strength", "Sport-Specific Training"]
+        )
+        equipment_available = st.multiselect(
+            "Equipment Available",
+            ["None (Bodyweight only)", "Dumbbells", "Resistance Bands", "Kettlebells", "Pull-up Bar", "Bench", "Full Gym Access"]
+        )
+        workout_days = st.text_input("Days Per Week", "3")
+        workout_location = st.radio("Workout Location", ["Home", "Gym"])
+        
+    with col2:
+        st.markdown("<div class='section-header'>Customize Prompt</div>", unsafe_allow_html=True)
+        
+        default_prompt = f"""
+You are an expert fitness coach creating a personalized weekly workout plan. 
+Primary goal: {primary_goal}
+Secondary goals: {', '.join(secondary_goals) if secondary_goals else 'None'}
+Fitness level: {fitness_level}
+Age: {user_age}
+Weight: {user_weight} kg
+Height: {user_height} cm
+Health conditions: {', '.join(health_conditions) if health_conditions else 'None'}
+Time available: {time_available} minutes per session
+Workout days: {workout_days} days per week
+Equipment: {', '.join(equipment_available) if equipment_available else 'None'}
+Location: {workout_location}
+
+Create a detailed weekly workout plan with these requirements:
+1. Include {workout_days} workout days with rest days appropriately spaced
+2. Each workout should be completable within {time_available} minutes
+3. Include general warm-up, then day-specific warmup
+4. Include general cool-down recommendations and warm-up sets for each exercise
+5. Provide specific exercises with sets, reps, rest periods, and recommended RPE 
+6. Use RPE and RIR based approach
+7. Use science based approach for exercise selection and progression
+8. Do not recommend redundant exercises (ie, 2 exercsies that target the same muscle, at the exact same angle and ROM)
+9. Include progression tips
+10. Do not give a specific day for core, recommend to do core exercises on every other day (max 3 exercises)
+11. All muscle groups need to be targeted 2x per week
+12. Format the response in a clear, organized way with days of the week as headers
+13. Include a brief explanation of why this plan suits the user's goals and fitness level
+14. Take into consideration that the user is at {fitness_level} fitness level
+"""
+        
+        prompt_text = st.text_area("Customize Prompt", default_prompt, height=400)
+        
+        if st.button("Generate Workout Plan", type="primary"):
+            if not st.session_state.openrouter_config["api_key"]:
+                st.error("Please enter your OpenRouter API key.")
+            else:
+                with st.spinner("Generating your personalized workout plan..."):
+                    try:
+                        model_config = MODELS[selected_model]
+                        
+                        headers = {
+                            "Authorization": f"Bearer {st.session_state.openrouter_config['api_key']}",
+                            "Content-Type": "application/json"
+                        }
+                        
+                        payload = {
+                            "model": model_config["model_id"],
+                            "messages": [
+                                {"role": "system", "content": "You are an expert fitness coach specializing in creating personalized workout plans."},
+                                {"role": "user", "content": prompt_text}
+                            ],
+                            "temperature": 0.7,
+                            "max_tokens": 4000
+                        }
+                        
+                        response = requests.post(
+                            "https://openrouter.ai/api/v1/chat/completions",
+                            headers=headers,
+                            data=json.dumps(payload)
+                        )
+                                            
+                        if response.status_code == 200:
+                            response_data = response.json()
+                        
+                            
+                            workout_plan = response_data["choices"][0]["message"]["content"]
+                            
+                            st.session_state.workout_plan = workout_plan
+                            st.session_state.generation_time = datetime.now().strftime("%Y-%m-%d %H:%M")
+                            st.session_state.model_used = selected_model
+                            
+                            st.success("Workout plan generated successfully!")
+                        else:
+                            error_details = f"Status Code: {response.status_code}"
+                            try:
+                                error_json = response.json()
+                                error_details += f"\nError: {json.dumps(error_json, indent=2)}"
+                            except:
+                                error_details += f"\nResponse Text: {response.text}"
+                            
+                            st.error(f"API Error: {error_details}")
+                    
+                    except Exception as e:
+                        st.error(f"An error occurred: {str(e)}")
+                        import traceback
+                        st.code(traceback.format_exc())
+
+
+    if st.session_state.workout_plan:
+        st.markdown("---")
+        st.markdown("<div class='section-header'>Your Personalized Workout Plan</div>", unsafe_allow_html=True)
+        
+        col1, col2 = st.columns([1, 1])
+        with col1:
+            st.info(f"Generated on: {st.session_state.generation_time}")
+        with col2:
+            st.info(f"Model: {st.session_state.model_used}")
+        st.markdown(st.session_state.workout_plan)
+
+    st.markdown("---")
+    st.markdown("© 2025 AI Health Coach | Powered by OpenRouter")
+
+
+def main():
+    if st.session_state.authenticated:
+        show_main_app()
+    else:
+        show_login_page()
+
+if __name__ == "__main__":
+    main()