logo

Author: Adarshagupta

app.py

@@ -51,9 +51,9 @@ def init_db():
 
 async def generate_chunk(api, model, topic, current_word_count, language, is_new_chapter=False):
     if is_new_chapter:
-        prompt = f"Write a detailed chapter for a book about {topic} in {language}. This is around word {current_word_count} of the book. Start with a chapter title, then write at least 500 words of content."
+        prompt = f"Write a detailed chapter for a book about {topic} in {language}. This is around word {current_word_count} of the book. Start with a chapter title, then write at least {current_word_count} words of content."
     else:
-        prompt = f"Continue writing a detailed book about {topic} in {language}. This is around word {current_word_count} of the book. Write at least 500 words, ensuring the narrative flows smoothly from the previous section."
+        prompt = f"Continue writing a detailed book about {topic} in {language}. This is around word {current_word_count} of the book. Write at least {current_word_count} words, ensuring the narrative flows smoothly from the previous section."
 
     try:
         if api == 'openai':
@@ -78,7 +78,7 @@ async def generate_chunk(api, model, topic, current_word_count, language, is_new
             response = together_client.chat.completions.create(
                 model=model,
                 messages=[
-                    {"role": "system", "content": f"You are an author writing a detailed book in {language}. Provide long, comprehensive responses with at least 500 words per chunk."},
+                    {"role": "system", "content": f"You are an author writing a detailed book in {language}. Provide long, comprehensive responses with at least {current_word_count} words per chunk."},
                     {"role": "user", "content": prompt}
                 ],
                 max_tokens=2000,
@@ -92,7 +92,7 @@ async def generate_chunk(api, model, topic, current_word_count, language, is_new
 
             # Ensure minimum word count
             while len(generated_text.split()) < 500:
-                additional_prompt = f"Continue the previous text, adding more details and expanding the narrative. Write at least {500 - len(generated_text.split())} more words."
+                additional_prompt = f"Continue the previous text, adding more details and expanding the narrative. Write at least {current_word_count} more words."
                 additional_response = together_client.chat.completions.create(
                     model=model,
                     messages=[

pdfs.db

@@ -94,15 +94,16 @@
         <div
             class="w-full lg:w-96 bg-white shadow-2xl p-6 lg:p-8 flex flex-col rounded-3xl transition-all duration-300 hover:shadow-3xl mobile-full-width glass-effect">
             <div class="flex justify-between items-center mb-6 lg:mb-10">
-                <img src="https://raw.githubusercontent.com/Adarshagupta/BookAI/main/logo.png" alt="BookAI Logo"
-                    class="w-12 h-10">
+               
                 <button id="profile-button" class="text-gray-600 hover:text-gray-800 mr-4">
                     <svg xmlns="http://www.w3.org/2000/svg" class="h-6 w-6" fill="none" viewBox="0 0 24 24"
                         stroke="currentColor">
                         <path stroke-linecap="round" stroke-linejoin="round" stroke-width="2"
                             d="M16 7a4 4 0 11-8 0 4 4 0 018 0zM12 14a7 7 0 00-7 7h14a7 7 0 00-7-7z" />
                     </svg>
                 </button>
+                <img src="https://raw.githubusercontent.com/Adarshagupta/BookAI/main/logo.png" alt="BookAI Logo"
+                class="w-12 h-10">
                 <button id="settings-button" class="text-gray-600 hover:text-gray-800">
                     <svg xmlns="http://www.w3.org/2000/svg" class="h-6 w-6" fill="none" viewBox="0 0 24 24"
                         stroke="currentColor">
@@ -133,7 +134,7 @@
                             <option value="gpt-4-turbo">GPT-4-turbo</option>
                             <option value="gpt-4o-mini-2024-07-18">GPT-4o Mini (2024-07-18)</option>
                         </optgroup>
-                        <optgroup label="Together Models" id="together-models">
+                        <optgroup label="BookAI Models" id="together-models">
                             <option value="meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo">BookAI-8B-Instruct-Turbo</option>
                             <option value="meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo">BookAI-70B-Instruct-Turbo</option>
                         </optgroup>