Made changes to Getting started with ALwrity and added lot of details on API keys

alwrity.py

@@ -1,16 +1,11 @@
 import streamlit as st
-import os
-import json
-import base64
-import logging
-from datetime import datetime
 
 # Set page config - must be the first Streamlit command
 st.set_page_config(
     page_title="AI Writer - Content Generation Platform",
     page_icon="✍️",
     layout="wide",
-    initial_sidebar_state="expanded",  # Changed from collapsed to expanded
+    initial_sidebar_state="collapsed",  # Start with collapsed sidebar
     menu_items={
         'Get Help': None,
         'Report a bug': None,
@@ -18,32 +13,27 @@ st.set_page_config(
     }
 )
 
-# Load and apply custom CSS
-with open('lib/workspace/alwrity_ui_styling.css', 'r') as f:
-    css = f.read()
-    
-st.markdown(f"""
+# Add CSS to hide sidebar during setup
+st.markdown("""
     <style>
-        /* Hide Streamlit header elements */
-        header {{
+        #MainMenu {visibility: hidden;}
+        footer {visibility: hidden;}
+        .stDeployButton {display:none;}
+        /* Hide sidebar during setup */
+        [data-testid="stSidebar"] {
             visibility: hidden !important;
-            height: 0px !important;
-        }}
-        
-        /* Hide Deploy button */
-        .stDeployButton {{
-            display: none !important;
-        }}
-        
-        /* Adjust top padding since we removed the header */
-        .main .block-container {{
-            padding-top: 1rem !important;
-        }}
-        
-        {css}
+            width: 0px !important;
+            position: fixed !important;
+        }
     </style>
 """, unsafe_allow_html=True)
 
+import os
+import json
+import base64
+import logging
+from datetime import datetime
+
 # Configure logging
 logging.basicConfig(
     level=logging.DEBUG,
@@ -55,11 +45,37 @@ logging.basicConfig(
 )
 logger = logging.getLogger(__name__)
 
+from lib.utils.config_manager import save_config
 from lib.utils.ui_setup import setup_ui
+from lib.utils.alwrity_sidebar import sidebar_configuration
 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.ui_setup import setup_ui, setup_alwrity_ui
+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
+from lib.utils.seo_tools import ai_seo_tools
+from lib.utils.ui_setup import setup_ui, setup_tabs
+from lib.utils.alwrity_utils import ai_agents_team, ai_social_writer
+from lib.utils.file_processor import load_image, read_prompts, write_prompts
+from lib.utils.voice_processing import record_voice
+
+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():
@@ -200,5 +216,52 @@ 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()