diff --git a/senju/ai_gen.py b/senju/ai_gen.py index 2b3294c..ff02baa 100644 --- a/senju/ai_gen.py +++ b/senju/ai_gen.py @@ -9,8 +9,10 @@ AI_GEN_SYS_PROMPT = """ You are a helpful AI agent whos task it is to generate Haikus from user input. Here is the definition of a Haiku: -Haiku (俳句) is a type of short form poetry that originated in Japan. Traditional Japanese haiku consist of three -phrases composed of 17 morae (called on in Japanese) in a 5, 7, 5 pattern that include a kireji, +Haiku (俳句) is a type of short form poetry that originated in Japan. +Traditional Japanese haiku consist of three +phrases composed of 17 morae (called on in Japanese) +in a 5, 7, 5 pattern that include a kireji, or "cutting word" and a kigo, or seasonal reference. The poem must contain the subjects of the user input. @@ -26,7 +28,8 @@ Your input is as follows: """ def request_haiku(seed: str) -> Haiku: - """This function prompts the ai to generate the hauku based on the user input""" + """This function prompts the ai to generate + the hauku based on the user input""" ai_gen_request = { "model": "llama3.2:1b", diff --git a/senju/main.py b/senju/main.py index b1641bb..e974ad1 100644 --- a/senju/main.py +++ b/senju/main.py @@ -1,7 +1,6 @@ from pathlib import Path from flask import Flask, redirect, render_template, url_for, request -from senju import haiku from senju.ai_gen import request_haiku from senju.haiku import Haiku from senju.store_manager import StoreManager