Google Grounded Search and Styling Fixes

alwrity.py

@@ -59,30 +59,9 @@ from lib.utils.ui_setup import setup_ui
 from lib.utils.api_key_manager.api_key_manager import APIKeyManager, render
 from lib.utils.api_key_manager.validation import check_all_api_keys
 from dotenv import load_dotenv
-from lib.utils.content_generators import blog_from_keyword, story_input_section, essay_writer, ai_news_writer, ai_finance_ta_writer, write_ai_prod_desc, do_web_research, competitor_analysis
 from lib.utils.ui_setup import setup_ui, setup_alwrity_ui
 
 
-def process_folder_for_rag(folder_path):
-    """Placeholder for the process_folder_for_rag function."""
-    logger.info(f"Processing folder for RAG: {folder_path}")
-    st.write(f"This is a placeholder for processing the folder: {folder_path}")
-
-
-def save_config(config):
-    """
-    Saves the provided configuration dictionary to a JSON file specified by the environment variable.
-    """
-    try:
-        logger.debug(f"Saving configuration to {os.getenv('ALWRITY_CONFIG')}")
-        with open(os.getenv("ALWRITY_CONFIG"), "w") as config_file:
-            json.dump(config, config_file, indent=4)
-        logger.info("Configuration saved successfully")
-    except Exception as e:
-        logger.error(f"Error saving configuration: {str(e)}", exc_info=True)
-        st.error(f"An error occurred while saving the configuration: {e}")
-
-
 def main():
     """Main application entry point."""
     # Initialize API key manager
@@ -221,52 +200,5 @@ def setup_environment_paths():
         raise
 
 
-# Functions for the main options
-def ai_writers():
-    options = [
-        "AI Blog Writer",
-        "Story Writer",
-        "Essay writer",
-        "Write News reports",
-        "Write Financial TA report",
-        "AI Product Description Writer",
-        "AI Copywriter",
-        "Quit"
-    ]
-    choice = st.selectbox("**👇Select a content creation type:**", options, index=0, format_func=lambda x: f"📝 {x}")
-
-    if choice == "AI Blog Writer":
-        blog_from_keyword()
-    elif choice == "Story Writer":
-        story_input_section()
-    elif choice == "Essay writer":
-        essay_writer()
-    elif choice == "Write News reports":
-        ai_news_writer()
-    elif choice == "Write Financial TA report":
-        ai_finance_ta_writer()
-    elif choice == "AI Product Description Writer":
-        write_ai_prod_desc()
-    elif choice == "Quit":
-        st.subheader("Exiting, Getting Lost. But.... I have nowhere to go 🥹🥹")
-
-
-
-def alwrity_brain():
-    st.title("🧠 Alwrity Brain, Better than yours!")
-    st.write("Choose a folder to write content on. Alwrity will do RAG on these documents. The documents can of any type, pdf, pptx, docs, txt, cs etc. Video files and Audio files are also permitted.")
-
-    folder_path = st.text_input("**Enter folder path:**")
-    if st.button("**Process Folder**"):
-        if folder_path:
-            try:
-                process_folder_for_rag(folder_path)
-                st.success("Folder processed successfully!")
-            except Exception as e:
-                st.error(f"Error processing folder: {e}")
-        else:
-            st.warning("Please enter a valid folder path.")
-
-
 if __name__ == "__main__":
     main()