Spaces:
Build error
Build error
| import os | |
| import streamlit as st | |
| import threading | |
| from streamlit.runtime.scriptrunner import add_script_run_ctx | |
| from classes.response_off import SeoOffPageAnalyst | |
| from classes.response_on_page import SeoOn | |
| from classes.response_website_and_tools import WebsiteAndTools | |
| from classes.response_seo import Seo | |
| from classes.response_social_media import SocialMedia | |
| def run_analysis(): | |
| # Placeholders for status updates | |
| off_page_status = st.empty() | |
| on_page_status = st.empty() | |
| website_and_tools_status = st.empty() | |
| seo_status = st.empty() | |
| social_media_status = st.empty() | |
| # Function to run SEO Off Page Analysis | |
| def run_off_page_analysis(): | |
| try: | |
| off_page_status.info("Starting SEO Off Page Analysis...") | |
| result = SeoOffPageAnalyst(os.getenv('MODEL_Off_Page_Analyst')) | |
| off_page_status.success("SEO Off Page Analysis completed successfully.") | |
| return result | |
| except Exception as e: | |
| off_page_status.error(f"SEO Off Page Analysis failed: {e}") | |
| return None | |
| # Function to run On Page Analysis | |
| def run_on_page_analysis(): | |
| try: | |
| on_page_status.info("Starting On Page Analysis...") | |
| result = SeoOn(os.getenv('MODEL_On_Page_Analyst')) | |
| on_page_status.success("On Page Analysis completed successfully.") | |
| return result | |
| except Exception as e: | |
| on_page_status.error(f"On Page Analysis failed: {e}") | |
| return None | |
| def run_website_and_tools_analysis(): | |
| try: | |
| website_and_tools_status.info("Starting Website and Tools Analysis...") | |
| result = WebsiteAndTools(os.getenv('Model_Website_and_Tools_Analyst')) | |
| website_and_tools_status.success("Website and Tools completed successfully.") | |
| return result | |
| except Exception as e: | |
| on_page_status.error(f"Website and Tools Analysis failed: {e}") | |
| return None | |
| def run_seo_analysis(): | |
| try: | |
| seo_status.info("Starting SEO Analysis...") | |
| result = Seo(os.getenv('MODEL_SEO_Analyst')) | |
| seo_status.success("SEO Analysis completed successfully.") | |
| return result | |
| except Exception as e: | |
| seo_status.error(f"SEO Analysis failed: {e}") | |
| return None | |
| def run_social_media_analysis(): | |
| try: | |
| social_media_status.info("Starting Social Media Analysis...") | |
| result = SocialMedia(os.getenv('MODEL_Social_Media_Analyst')) | |
| social_media_status.success("Social Media Analysis completed successfully.") | |
| return result | |
| except Exception as e: | |
| social_media_status.error(f"Social Media Analysis failed: {e}") | |
| return None | |
| # Create threads for concurrent execution | |
| off_page_thread = threading.Thread(target=run_off_page_analysis) | |
| on_page_thread = threading.Thread(target=run_on_page_analysis) | |
| website_and_tools_thread = threading.Thread(target=run_website_and_tools_analysis) | |
| seo_thread = threading.Thread(target=run_seo_analysis) | |
| social_media_thread = threading.Thread(target=run_social_media_analysis) | |
| # Attach Streamlit context to threads | |
| add_script_run_ctx(off_page_thread) | |
| add_script_run_ctx(on_page_thread) | |
| add_script_run_ctx(website_and_tools_thread) | |
| add_script_run_ctx(seo_thread) | |
| add_script_run_ctx(social_media_thread) | |
| # Start threads | |
| off_page_thread.start() | |
| on_page_thread.start() | |
| website_and_tools_thread.start() | |
| seo_thread.start() | |
| social_media_thread.start() | |
| # Wait for threads to complete | |
| off_page_thread.join() | |
| on_page_thread.join() | |
| website_and_tools_thread.join() | |
| seo_thread.join() | |
| st.success("🎉 All analyses completed!") # Final success message | |
| # --- Display Button After Completion --- | |
| if st.button("View Results"): | |
| st.switch_page("pages/output.py") | |
| # Execute the analysis | |
| if st.button("Back"): | |
| st.switch_page("pages/home.py") | |
| run_analysis() | |