Upload folder using huggingface_hub

book_generator.py

@@ -7,13 +7,13 @@ from typing import Dict
 import gradio as gr  # Import Gradio
 
 # Step 3: Set the environment variable for the Groq API Key
-os.environ["GROQ_API_KEY"] = os.getenv("GROQ_API_KEY")  # Use environment variable for API Key
+os.environ["GROQ_API_KEY"] = "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"  # Use provided secret key
 
 # Step 4: Define helper functions for structured book generation
 def create_book_agent(
     model_name: str = os.getenv("MODEL_NAME", "llama-3.1-8b-instant"),  # Use environment variable for model name
     temperature: float = 0.7,
-    max_tokens: int = 16384,  # Increased token limit
+    max_tokens: int = 8000,  # Set max_tokens to 8000
     **kwargs
 ) -> ChatGroq:
     """Create a LangChain agent for book writing."""
@@ -21,41 +21,37 @@ def create_book_agent(
         ("system", "You are a creative writer. Write high-quality, engaging books for any genre."),
         ("human", "{input}")
     ])
-    token = os.getenv("HUGGINGFACE_TOKEN")  # Get Hugging Face token from environment variable
-    llm = ChatGroq(model=model_name, temperature=temperature, max_tokens=max_tokens, token=token, **kwargs)
+    llm = ChatGroq(model=model_name, temperature=temperature, max_tokens=max_tokens, **kwargs)  # Removed token parameter
     chain = prompt_template | llm | StrOutputParser()
     return chain
 
-def generate_chapter(title: str, synopsis: str, agent) -> str:
-    """Generate a full chapter given a title and synopsis."""
-    query = f"Write a detailed chapter based on the following synopsis:\n\nTitle: {title}\n\nSynopsis: {synopsis}"
+def generate_chapter(title: str, agent) -> str:
+    """Generate a full chapter given a title."""
+    query = f"Write a detailed chapter titled '{title}'"
     try:
         return agent.invoke({"input": query})
     except Exception as e:
         print(f"An error occurred while generating the chapter: {e}")
         return ""
 
-def write_book(agent, title: str, outline: Dict[str, str]) -> str:
+def write_book(title: str, outline: Dict[str, str]) -> str:
     """
     Generate a complete book.
     
     Args:
-        agent: The LangChain agent for generating text.
         title (str): The title of the book.
-        outline (Dict[str, str]): A dictionary with chapter titles as keys and synopses as values.
+        outline (Dict[str, str]): A dictionary with chapter titles as keys.
 
     Returns:
         str: The full book as a single string.
     """
     book = f"# {title}\n\n"
-    for chapter_title, chapter_synopsis in outline.items():
+    for chapter_title in outline.keys():
         book += f"## {chapter_title}\n\n"
-        chapter_text = generate_chapter(chapter_title, chapter_synopsis, agent)
+        agent = create_book_agent()  # Create a new agent for each chapter
+        chapter_text = generate_chapter(chapter_title, agent)
         book += chapter_text + "\n\n"
-    return book
-
-# Step 5: Create the agent
-book_agent = create_book_agent()
+    return book  # Return the complete book
 
 # Step 6: Gradio interface
 def gradio_interface(api_key: str = ""):
@@ -63,30 +59,62 @@ def gradio_interface(api_key: str = ""):
     with gr.Blocks() as demo:
         gr.Markdown("## Book Generator")
         gr.Markdown("This application was created by iLL-Ai AaronAllton and a team of Groq agents that write books.")  # Updated note
-        book_title = gr.Textbox(label="Book Title")
-        book_outline = gr.Textbox(label="Book Outline (Structured format, e.g., 'Chapter 1: Synopsis 1; Chapter 2: Synopsis 2')")  # Updated prompt
-        generate_button = gr.Button("Generate Book")
+        book_title = gr.Textbox(label="Book Title", placeholder="Enter the title of your book")  # Added placeholder
+        book_outline = gr.Textbox(label="Number of Chapters", placeholder="e.g., '10'", info="Specify the number of chapters you want in your book.")  # Added info
+        api_key_input = gr.Textbox(label="API Key", type="password", placeholder="Enter your API key")  # Added placeholder
+        book_description = gr.Textbox(label="Book Description", placeholder="Enter a brief description of the book")  # New description textbox
+        book_type = gr.Dropdown(label="Type of Book", choices=[
+            "Erotic Novel", 
+            "Step-by-Step Tutorial", 
+            "Autobiography", 
+            "Biography", 
+            "Novel", 
+            "Horror", 
+            "Science Fiction", 
+            "Fantasy", 
+            "Mystery", 
+            "Thriller", 
+            "Romance", 
+            "Historical Fiction", 
+            "Self-Help", 
+            "Cookbook", 
+            "Children's Book", 
+            "Graphic Novel"
+        ], value="Novel", info="Select the genre or type of book you want to create.")  # Added info
+        generate_button = gr.Button("Generate Outline")
         output = gr.Textbox(label="Generated Book", interactive=False)
 
-        def generate_book_interface(title, outline):
+        def generate_book_interface(title, outline, api_key):  # Add api_key parameter
             try:
                 # Normalize the outline input
                 outline_dict = {}
-                chapters = outline.split(';')  # Split by semicolon for each chapter
-                for chapter in chapters:
-                    if ':' in chapter:
-                        title, synopsis = chapter.split(':', 1)
-                        outline_dict[title.strip()] = synopsis.strip()
-                    else:
-                        # Handle cases where the input might not follow the expected format
-                        outline_dict[chapter.strip()] = "No synopsis provided."
+                if not outline.strip():  # Check if outline is empty
+                    return "Please provide a valid outline."  # Prompt user for valid outline
+                
+                # Generate outline based on user input
+                num_chapters = int(outline)  # Extract number of chapters
+                for i in range(1, num_chapters + 1):
+                    outline_dict[f"Chapter {i}"] = None  # No synopsis needed
                 
                 print(f"Processed Outline: {outline_dict}")  # Debug statement
-                return write_book(book_agent, title, outline_dict)
+                agent = create_book_agent(api_key)  # Create agent with user-provided API key
+                return write_book(title, outline_dict)  # Call the generator
             except Exception as e:
                 return f"An error occurred: {e}"
 
-        generate_button.click(generate_book_interface, inputs=[book_title, book_outline], outputs=output)
+        generate_button.click(generate_book_interface, inputs=[book_title, book_outline, api_key_input], outputs=output)
+
+        # New button to generate the book from the outline
+        generate_book_button = gr.Button("Generate Book")
+        generate_book_output = gr.Textbox(label="Final Book", interactive=False)
+
+        def generate_final_book(title, outline):
+            if not outline.strip():
+                return "Please provide a valid outline."
+            outline_dict = {title: outline}
+            return write_book(title, outline_dict)
+
+        generate_book_button.click(generate_final_book, inputs=[book_title, book_outline], outputs=generate_book_output)
 
     demo.launch(share=True)  # Enable sharing of the Gradio app
 

button_functionality_overview.md

@@ -0,0 +1,29 @@
+# Button Functionality Overview for the Book Generator
+
+## 1. Generate Outline Button
+- **Description**: Initiates the outline generation process based on user input.
+- **Functionality**: Takes the provided book title and number of chapters to generate an outline for the book.
+
+## 2. Generate Book Button
+- **Description**: Generates the complete book based on the outline.
+- **Functionality**: Takes the book title and outline to create the full book content.
+
+## 3. Clear Button
+- **Description**: Clears all input fields.
+- **Functionality**: Resets the application to its initial state, allowing the user to start fresh.
+
+## 4. Save Button
+- **Description**: Saves the generated book content.
+- **Functionality**: Allows the user to download or save the content to a specified location.
+
+## 5. Load Button
+- **Description**: Loads a previously saved book.
+- **Functionality**: Enables the user to import existing book content for editing or continuation.
+
+## 6. Settings Button
+- **Description**: Opens the settings menu.
+- **Functionality**: Allows the user to adjust application settings, such as model selection or output format.
+
+## 7. Help Button
+- **Description**: Provides assistance or documentation.
+- **Functionality**: Displays help information or guides on how to use the application.

outline_for_iLLAi_book.md

@@ -0,0 +1,30 @@
+# Outline for iLLAi-book
+
+## Title
+- iLLAi-book
+
+## Introduction
+- Brief overview of the project and its purpose.
+
+## Requirements
+- List of dependencies and environment variables needed to run the application.
+  - `GROQ_API_KEY`: Your Groq API key.
+  - `MODEL_NAME`: The name of the model to be used (optional, defaults to "llama-3.1-8b-instant").
+
+## Installation
+- Instructions on how to set up the project locally.
+
+## Usage
+- How to run the application and any relevant commands.
+
+## Features
+- Key functionalities of the book generator.
+
+## Contributing
+- Guidelines for contributing to the project.
+
+## License
+- Information about the project's licensing.
+
+## Contact
+- How to reach the maintainers for questions or support.