kunthawat/ALwrity
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: Improve import organization, add error handling for file operations, and enhance code robustness in alwrity.py

Browse Source
This commit is contained in:
ajaysi (aider)
2024-09-14 17:30:27 +05:30
parent 719b863d95
commit dded2180f3
1 changed files with 15 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
alwrity.py
View File

@@ -1,19 +1,17 @@
import os import os
import json import json
import base64
from datetime import datetime from datetime import datetime
from dotenv import load_dotenv from dotenv import load_dotenv
import streamlit as st import streamlit as st
import base64
# Load .env file # Load .env file
load_dotenv() load_dotenv()
#from lib.chatbot_custom.chatbot_local_docqa import alwrity_chat_docqa 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, blog_from_keyword, ai_agents_team, essay_writer, ai_news_writer, ai_seo_tools,
ai_finance_ta_writer, ai_social_writer, ai_finance_ta_writer, ai_social_writer, do_web_research, competitor_analysis
do_web_research, competitor_analysis, )
)
from lib.ai_writers.ai_story_writer.story_writer import story_input_section from lib.ai_writers.ai_story_writer.story_writer import story_input_section
from lib.ai_writers.ai_product_description_writer import write_ai_prod_desc from lib.ai_writers.ai_product_description_writer import write_ai_prod_desc
from lib.content_planning_calender.content_planning_agents_alwrity_crew import ai_agents_content_planner from lib.content_planning_calender.content_planning_agents_alwrity_crew import ai_agents_content_planner
@@ -51,8 +49,11 @@ def check_api_keys():
api_key = st.text_input(f"Enter 🔏 {key}: 👉[Get it here]({url})👈") api_key = st.text_input(f"Enter 🔏 {key}: 👉[Get it here]({url})👈")
if api_key: if api_key:
os.environ[key] = api_key os.environ[key] = api_key
with open(".env", "a") as env_file: try:
env_file.write(f"{key}={api_key}\n") with open(".env", "a") as env_file:
env_file.write(f"{key}={api_key}\n")
except IOError as e:
st.error(f"Failed to write {key} to .env file: {e}")
st.success(f"✅ {key} added successfully!") st.success(f"✅ {key} added successfully!")
return False return False
return True return True
@@ -75,8 +76,11 @@ def check_llm_environs():
"Select your LLM Provider", options=list(supported_providers.keys()) "Select your LLM Provider", options=list(supported_providers.keys())
) )
os.environ["GPT_PROVIDER"] = gpt_provider os.environ["GPT_PROVIDER"] = gpt_provider
with open(".env", "a") as env_file: try:
env_file.write(f"GPT_PROVIDER={gpt_provider}\n") with open(".env", "a") as env_file:
env_file.write(f"GPT_PROVIDER={gpt_provider}\n")
except IOError as e:
st.error(f"Failed to write GPT_PROVIDER to .env file: {e}")
st.success(f"GPT Provider set to {gpt_provider}") st.success(f"GPT Provider set to {gpt_provider}")
api_key_var = supported_providers[gpt_provider.lower()] api_key_var = supported_providers[gpt_provider.lower()]