Create app.py

app.py

@@ -0,0 +1,130 @@
+import streamlit as st
+import os
+import sys
+from crewai import Agent, Task, Crew, Process
+from crewai_tools import SerperDevTool, BaseTool
+
+# 1. Page Configuration
+st.set_page_config(page_title="Smart Relocation Assistant", page_icon="🏘️")
+
+st.title("🏘️ Smart Relocation Assistant")
+st.markdown("Use AI Agents to find your perfect neighborhood based on budget and lifestyle.")
+
+# 2. Sidebar for API Keys (Optional but recommended for public spaces)
+with st.sidebar:
+    st.header("API Configuration")
+    st.info("If running locally, set keys in .env. On Hugging Face, set them in 'Settings' -> 'Secrets'.")
+    
+    # Check if keys are already in environment (HF Secrets)
+    openai_api_key = os.getenv("OPENAI_API_KEY")
+    serper_api_key = os.getenv("SERPER_API_KEY")
+
+    # If not found in env, ask user to input them
+    if not openai_api_key:
+        openai_api_key = st.text_input("OpenAI API Key", type="password")
+    if not serper_api_key:
+        serper_api_key = st.text_input("Serper API Key", type="password")
+
+# 3. Define Tools
+class CalculatorTool(BaseTool):
+    name: str = "Budget Calculator"
+    description: str = "Useful for calculating if a rent price fits within a budget."
+
+    def _run(self, expression: str) -> str:
+        try:
+            return str(eval(expression))
+        except:
+            return "Error in calculation"
+
+# 4. Main Application Logic
+if openai_api_key and serper_api_key:
+    # Set keys for the libraries
+    os.environ["OPENAI_API_KEY"] = openai_api_key
+    os.environ["SERPER_API_KEY"] = serper_api_key
+    
+    # Initialize Tools
+    search_tool = SerperDevTool()
+    calc_tool = CalculatorTool()
+
+    # Inputs
+    col1, col2 = st.columns(2)
+    with col1:
+        city = st.text_input("Target City", "Austin, TX")
+        budget = st.number_input("Max Monthly Rent ($)", min_value=500, value=2500)
+    with col2:
+        work_location = st.text_input("Work Location", "Downtown")
+        
+    preferences = st.text_area("Lifestyle Preferences", "Walkable, near coffee shops, safe, good for young professionals")
+
+    if st.button("Start Research"):
+        with st.spinner('🤖 AI Agents are scouting neighborhoods and crunching numbers...'):
+            try:
+                # --- AGENT DEFINITIONS ---
+                city_scout = Agent(
+                    role='City Neighborhood Scout',
+                    goal=f'Identify the top 3 neighborhoods in {city} that match: {preferences}.',
+                    backstory="You are a local expert who knows every corner of the city. You prioritize safety, commute, and lifestyle fit.",
+                    tools=[search_tool],
+                    verbose=True,
+                    allow_delegation=False
+                )
+
+                budget_analyst = Agent(
+                    role='Relocation Financial Advisor',
+                    goal=f'Assess affordability and ensure rent is within {budget}.',
+                    backstory="You are a pragmatic financial advisor. You find average rent prices and filter out options that exceed the budget.",
+                    tools=[search_tool, calc_tool],
+                    verbose=True,
+                    allow_delegation=False
+                )
+
+                # --- TASK DEFINITIONS ---
+                scout_task = Task(
+                    description=f"""
+                    1. Search for neighborhoods in {city} that are: {preferences}.
+                    2. Consider commute to {work_location}.
+                    3. Select top 5 candidates.
+                    """,
+                    expected_output="List of 5 neighborhoods with vibe descriptions.",
+                    agent=city_scout
+                )
+
+                analysis_task = Task(
+                    description=f"""
+                    1. For the neighborhoods found, search current average rent for 1-bedroom.
+                    2. Filter out any where rent > {budget}.
+                    3. Calculate remaining budget.
+                    """,
+                    expected_output="Filtered list of affordable neighborhoods with costs.",
+                    agent=budget_analyst
+                )
+
+                report_task = Task(
+                    description="""
+                    Compile a final report. Rank top 3 affordable neighborhoods.
+                    Include: Neighborhood Name, Vibe, Rent Cost, Remaining Budget, and one Local Highlight.
+                    Format as clear Markdown.
+                    """,
+                    expected_output="Markdown formatted relocation guide.",
+                    agent=budget_analyst
+                )
+
+                # --- CREW EXECUTION ---
+                relocation_crew = Crew(
+                    agents=[city_scout, budget_analyst],
+                    tasks=[scout_task, analysis_task, report_task],
+                    process=Process.sequential
+                )
+
+                result = relocation_crew.kickoff()
+                
+                # Display Results
+                st.success("Research Complete!")
+                st.markdown("### 📋 Your Relocation Plan")
+                st.markdown(result)
+                
+            except Exception as e:
+                st.error(f"An error occurred: {e}")
+
+else:
+    st.warning("Please provide your API Keys to proceed.")