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

Alwrity AI Essay writer

Browse Source
This commit is contained in:
AjaySi
2024-04-14 17:54:28 +05:30
parent 31995df984
commit 80c52facc5
3 changed files with 241 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

55
alwrity.py
View File

@@ -7,6 +7,10 @@ from prompt_toolkit.shortcuts import checkboxlist_dialog, message_dialog, input_
from prompt_toolkit import prompt
from prompt_toolkit.styles import Style
from prompt_toolkit.shortcuts import radiolist_dialog
from prompt_toolkit.formatted_text import HTML
from prompt_toolkit.layout.containers import HSplit, Window
from prompt_toolkit.layout.controls import BufferControl
from dotenv import load_dotenv
import requests
from rich import print
@@ -21,6 +25,7 @@ from lib.ai_web_researcher.metaphor_basic_neural_web_search import metaphor_find
from lib.ai_writers.keywords_to_blog import write_blog_from_keywords
from lib.ai_writers.speech_to_blog.main_audio_to_blog import generate_audio_blog
from lib.gpt_providers.text_generation.ai_story_writer import ai_story_generator
from lib.gpt_providers.text_generation.ai_essay_writer import ai_essay_generator
def prompt_for_time_range():
@@ -60,6 +65,7 @@ def start_interactive_mode():
choices = [
("AI Blog Writer", "AI Blog Writer"),
("AI Story Writer", "AI Story Writer"),
("AI Essay Writer", "AI Essay Writer"),
("Do keyword Research", "Do keyword Research"),
("Create Blog Images(TBD)", "Create Blog Images(TBD)"),
("Competitor Analysis", "Competitor Analysis"),
@@ -73,6 +79,8 @@ def start_interactive_mode():
write_blog()
elif mode == 'AI Story Writer':
write_story()
elif mode == 'AI Essay Writer':
essay_writer()
elif mode == 'Do keyword Research':
do_web_research()
elif mode == 'Create Blog Images(TBD)':
@@ -139,7 +147,7 @@ def get_api_key(api_key: str, api_description: str):
api_key (str): The name of the API key variable.
api_description (str): The description of the API key.
"""
user_input = typer.prompt(f"\n🙆🙆Please enter {api_key} API Key:")
user_input = typer.prompt(f"\n\n🙆💩💩🙆 - Please enter {api_key} API Key:")
with open(".env", "a") as env_file:
env_file.write(f"{api_key}={user_input}\n")
print(f"✅ {api_description} API Key added to .env file.")
@@ -204,6 +212,51 @@ def write_story():
exit(1)
def essay_writer():
# Define essay types and education levels
essay_types = [
("Argumentative", "Argumentative - Forming an opinion via research. Building an evidence-based argument."),
("Expository", "Expository - Knowledge of a topic. Communicating information clearly."),
("Narrative", "Narrative - Creative language use. Presenting a compelling narrative."),
("Descriptive", "Descriptive - Creative language use. Describing sensory details.")
]
education_levels = [
("Primary School", "Primary School"),
("High School", "High School"),
("College", "College"),
("Graduate School", "Graduate School")
]
# Define the options for number of pages
num_pages_options = [
("Short Form (1-2 pages)", "Short Form"),
("Medium Form (3-5 pages)", "Medium Form"),
("Long Form (6+ pages)", "Long Form")
]
# Ask the user for the title of the essay
essay_title = input_dialog(title="Essay Title", text="Enter the title of your essay:").run()
while not essay_title.strip():
print("Please enter a valid title for your essay.")
essay_title = input_dialog(title="Essay Title", text="Enter the title of your essay:").run()
# Ask the user for type of essay, level of education, and number of pages
selected_essay_type = radiolist_dialog(title="Type of Essay", text="Choose the type of essay you want to write:",
values=essay_types).run()
selected_education_level = radiolist_dialog(title="Level of Education", text="Choose your level of education:",
values=education_levels).run()
# Prompt the user to select the length of the essay
num_pages_prompt = "Select the length of your essay:"
selected_num_pages = radiolist_dialog(title="Number of Pages", text=num_pages_prompt, values=num_pages_options).run()
ai_essay_generator(essay_title, selected_essay_type, selected_education_level, selected_num_pages)
def blog_tools():
os.system("clear" if os.name == "posix" else "cls")
text = "_______________________________________________________________________\n"

186
lib/gpt_providers/text_generation/ai_essay_writer.py Normal file
View File

@@ -0,0 +1,186 @@
#####################################################
#
# Alwrity, AI essay writer - Essay_Writing_with_Prompt_Chaining
#
#####################################################
import os
from pathlib import Path
from dotenv import load_dotenv
from google.api_core import retry
import google.generativeai as genai
from pprint import pprint
def generate_with_retry(model, prompt):
"""
Generates content from the model with retry handling for errors.
Parameters:
model (GenerativeModel): The generative model to use for content generation.
prompt (str): The prompt to generate content from.
Returns:
str: The generated content.
"""
try:
# FIXME: Need a progress bar here.
return model.generate_content(prompt, request_options={'retry':retry.Retry()})
except Exception as e:
print(f"Error generating content: {e}")
return ""
def ai_essay_generator(essay_title, selected_essay_type, selected_education_level, selected_num_pages):
"""
Write an Essay using prompt chaining and iterative generation.
Parameters:
persona (str): The persona statement for the author.
story_genre (str): The genre of the story.
characters (str): The characters in the story.
"""
print(f"Starting to write Essay on {essay_title}..")
try:
# Define persona and writing guidelines
guidelines = f'''\
Writing Guidelines
As an expert Essay writer and academic researcher, demostrate your world class essay writing skills.
Follow the below writing guidelines for writing your essay:
1). You specialize in {selected_essay_type} essay writing.
2). Your target audiences include readers from {selected_education_level} level.
3). The title of the essay is {essay_title}.
4). I will provide you with web research for essay title.
5). The final essay should of {selected_num_pages} words/pages.
3). Plant the seeds of subplots or potential character arc shifts that can be expanded later.
Remember, your main goal is to write as much as you can. If you get through
the story too fast, that is bad. Expand, never summarize.
'''
# Generate prompts
premise_prompt = f'''\
As an expert essay writer, specilizing in {selected_essay_type} essay writing.
Write an Essay title for given keywords {essay_title}.
The title should appeal to audience level of {selected_education_level}.
'''
outline_prompt = f'''\
As an expert essay writer, specilizing in {selected_essay_type} essay writing.
Your Essay title is:
{{premise}}
Write an outline for the essay.
'''
starting_prompt = f'''\
As an expert essay writer, specilizing in {selected_essay_type} essay writing.
Your essay title is:
{{premise}}
The outline of the Essay is:
{{outline}}
First, silently review the outline and the essay title. Consider how to start the Essay.
Start to write the very beginning of the Essay. You are not expected to finish
the whole Essay now. Your writing should be detailed enough that you are only
scratching the surface of the first bullet of your outline. Try to write AT
MINIMUM 1000 WORDS.
{guidelines}
'''
continuation_prompt = f'''\
As an expert essay writer, specilizing in {selected_essay_type} essay writing.
Your essay title is:
{{premise}}
The outline of the Essay is:
{{outline}}
You've begun to write the essay and continue to do so.
Here's what you've written so far:
{{story_text}}
=====
First, silently review the outline and essay so far.
Identify what the single next part of your outline you should write.
Your task is to continue where you left off and write the next part of the Essay.
You are not expected to finish the whole essay now. Your writing should be
detailed enough that you are only scratching the surface of the next part of
your outline. Try to write AT MINIMUM 1000 WORDS. However, only once the essay
is COMPLETELY finished, write IAMDONE. Remember, do NOT write a whole chapter
right now.
{guidelines}
'''
# Configure generative AI
load_dotenv(Path('../.env'))
genai.configure(api_key=os.getenv('GEMINI_API_KEY'))
# Initialize the generative model
model = genai.GenerativeModel('gemini-1.0-pro')
# Generate prompts
try:
premise = generate_with_retry(model, premise_prompt).text
print(f"The title of the Essay is: {premise}")
except Exception as err:
print(f"Essay title Generation Error: {err}")
return
outline = generate_with_retry(model, outline_prompt.format(premise=premise)).text
print(f"The Outline of the essay is: {outline}\n\n")
if not outline:
print("Failed to generate Essay outline. Exiting...")
return
try:
starting_draft = generate_with_retry(model,
starting_prompt.format(premise=premise, outline=outline)).text
pprint(starting_draft)
except Exception as err:
print(f"Failed to Generate Essay draft: {err}")
return
try:
draft = starting_draft
continuation = generate_with_retry(model,
continuation_prompt.format(premise=premise, outline=outline, story_text=draft)).text
pprint(continuation)
except Exception as err:
print(f"Failed to write the initial draft: {err}")
# Add the continuation to the initial draft, keep building the story until we see 'IAMDONE'
try:
draft += '\n\n' + continuation
except Exception as err:
print(f"Failed as: {err} and {continuation}")
while 'IAMDONE' not in continuation:
try:
continuation = generate_with_retry(model,
continuation_prompt.format(premise=premise, outline=outline, story_text=draft)).text
draft += '\n\n' + continuation
except Exception as err:
print(f"Failed to continually write the Essay: {err}")
return
# Remove 'IAMDONE' and print the final story
final = draft.replace('IAMDONE', '').strip()
pprint(final)
except Exception as e:
print(f"Main Essay writing: An error occurred: {e}")

2
lib/gpt_providers/text_generation/ai_story_writer.py
View File

@@ -42,7 +42,7 @@ def ai_story_generator(persona, story_genre, characters):
print(f"Starting to write {story_genre} story based on characters: {characters}..")
try:
# Define persona and writing guidelines
guidelines = '''\
guidelines = f'''\
Writing Guidelines
Delve deeper. Lose yourself in the world you're building. Unleash vivid