From 668e7f7f36c94e4cd1d34ac13cc631f7b45d285b Mon Sep 17 00:00:00 2001
From: "ajaysi (aider)" <ajay.calsoft@gmail.com>
Date: Sat, 14 Sep 2024 17:49:55 +0530
Subject: [PATCH] refactor: Fix undefined names by adding missing imports in
 alwrity.py, alwrity_utils.py, content_generators.py, and ui_setup.py

---
 alwrity.py                      |  7 ++++++-
 lib/utils/alwrity_utils.py      | 14 ++++++++++++++
 lib/utils/content_generators.py |  2 +-
 lib/utils/ui_setup.py           |  2 ++
 4 files changed, 23 insertions(+), 2 deletions(-)

diff --git a/alwrity.py b/alwrity.py
index 4fa2fa7d..a348e64a 100644
--- a/alwrity.py
+++ b/alwrity.py
@@ -1,9 +1,14 @@
 import streamlit as st
+import os
+import json
+import base64
+from datetime import datetime
+import streamlit as st
 from lib.utils.environment_utils import load_environment
 from lib.utils.config_manager import save_config
 from lib.utils.ui_setup import setup_ui, setup_tabs
 from lib.utils.api_key_manager import check_api_keys, check_llm_environs
-from lib.utils.content_generators import ai_writers, content_planning_tools
+from lib.utils.content_generators import ai_writers, content_planning_tools, blog_from_keyword, story_input_section, essay_writer, ai_news_writer, ai_finance_ta_writer, write_ai_prod_desc, do_web_research, competitor_analysis, ai_agents_content_planner
 from lib.utils.seo_tools import ai_seo_tools
 from lib.utils.file_processor import load_image, read_prompts, write_prompts
 from lib.utils.voice_processing import record_voice
diff --git a/lib/utils/alwrity_utils.py b/lib/utils/alwrity_utils.py
index c85adfc1..cdeab67e 100644
--- a/lib/utils/alwrity_utils.py
+++ b/lib/utils/alwrity_utils.py
@@ -18,7 +18,21 @@ from lib.ai_writers.youtube_ai_writer import write_yt_title, write_yt_descriptio
 from lib.ai_writers.web_url_ai_writer import blog_from_url
 from lib.ai_writers.image_ai_writer import blog_from_image
 from lib.ai_writers.ai_essay_writer import ai_essay_generator
+import os
+import PyPDF2
+import tiktoken
+import openai
 from lib.gpt_providers.text_to_image_generation.main_generate_image_from_prompt import generate_image
+from lib.ai_seo_tools.seo_structured_data import ai_structured_data
+from lib.ai_seo_tools.content_title_generator import ai_title_generator
+from lib.ai_seo_tools.meta_desc_generator import metadesc_generator_main
+from lib.ai_seo_tools.image_alt_text_generator import alt_text_gen
+from lib.ai_seo_tools.opengraph_generator import og_tag_generator
+from lib.ai_seo_tools.optimize_images_for_upload import main_img_optimizer
+from lib.ai_seo_tools.google_pagespeed_insights import google_pagespeed_insights
+from lib.ai_seo_tools.on_page_seo_analyzer import analyze_onpage_seo
+from lib.ai_seo_tools.weburl_seo_checker import url_seo_checker
+from lib.utils.voice_processing import record_voice
 from lib.content_planning_calender.content_planning_agents_alwrity_crew import ai_agents_content_planner
 from ..gpt_providers.text_generation.main_text_generation import llm_text_gen
 
diff --git a/lib/utils/content_generators.py b/lib/utils/content_generators.py
index 566a93be..27e639a5 100644
--- a/lib/utils/content_generators.py
+++ b/lib/utils/content_generators.py
@@ -1,5 +1,5 @@
 import streamlit as st
-from lib.utils.alwrity_utils import (
+from lib.utils.alwrity_utils import blog_from_keyword, ai_agents_team, essay_writer, ai_news_writer, ai_seo_tools, ai_finance_ta_writer, ai_social_writer, do_web_research, competitor_analysis, blog_from_audio
     blog_from_keyword, ai_agents_team, essay_writer, ai_news_writer, ai_seo_tools,
     ai_finance_ta_writer, ai_social_writer, do_web_research, competitor_analysis
 )
diff --git a/lib/utils/ui_setup.py b/lib/utils/ui_setup.py
index f9f51a4c..bddbdf3b 100644
--- a/lib/utils/ui_setup.py
+++ b/lib/utils/ui_setup.py
@@ -1,6 +1,8 @@
 import os
 import streamlit as st
 from .file_processor import load_image
+from .content_generators import content_planning_tools, ai_writers
+from .alwrity_utils import ai_agents_team, ai_seo_tools, ai_social_writer
 
 def setup_ui():
     """Sets up the Streamlit UI with custom CSS and logo."""