Update app.py

app.py

@@ -44,7 +44,6 @@ def scrape_website(url):
         return f"Failed to retrieve page. Status code: {response.status_code}"
 
 
-
 # Load environment variables from .env
 load_dotenv()
 
@@ -80,6 +79,12 @@ if uploaded_file is not None:
             "Real Estate",
             "Utilities"
         ]
+    
+    # Create directories if they don't exist
+    os.makedirs(Config.STOCK_DATA_DIR, exist_ok=True)
+    os.makedirs(os.path.dirname(Config.OUTPUT_FILE), exist_ok=True)
+    os.makedirs(os.path.dirname(Config.SCENARIO_OUTPUT_FILE), exist_ok=True)
+
     def configure_generative_ai():
         """Configures the generative AI model and starts a chat session."""
         genai.configure(api_key=Config.GOOGLE_API_KEY)
@@ -150,7 +155,6 @@ if uploaded_file is not None:
         scenario_response = get_response(chat_session, scenario_prompt)
         json_output = extract_json_content(scenario_response.text)
         scenario_data = json.loads(json_output)
-        os.makedirs(os.path.dirname(Config.SCENARIO_OUTPUT_FILE), exist_ok=True)
         with open(Config.SCENARIO_OUTPUT_FILE, "w") as f:
             json.dump(scenario_data, f, indent=4)
         st.success("Market scenarios saved successfully!")
@@ -163,6 +167,7 @@ if uploaded_file is not None:
 
     # Save simulation results
     simulation_results_file = "output_files/simulation_results.json"
+    os.makedirs(os.path.dirname(simulation_results_file), exist_ok=True)
     with open(simulation_results_file, "w") as file:
         json.dump(simulation_results, file, indent=4)
     st.success("Monte Carlo Simulation completed!")
@@ -171,4 +176,4 @@ if uploaded_file is not None:
     st.write("Download Output Files")
     for file in [Config.OUTPUT_FILE, Config.SCENARIO_OUTPUT_FILE, simulation_results_file]:
         with open(file, "rb") as f:
-            st.download_button(label=f"Download {file.split('/')[-1]}", data=f, file_name=file.split('/')[-1])
+            st.download_button(label=f"Download {file.split('/')[-1]}", data=f, file_name=file.split('/')[-1])