import os
from crewai import Agent, Task, Crew, Process, LLM
import gradio as gr
from markdown import markdown
import warnings
warnings.filterwarnings('ignore')

api_key = os.getenv("api_key")
llm = LLM(
              model='gemini/gemini-1.5-flash',
              api_key=api_key
            )


# Agents
def generate_fantasy_outline(user_idea:str):
    idea_extractor = Agent(
        role="Idea Extractor",
        goal=f"Extract and refine a raw fantasy story idea into structured elements from {user_idea}.",
        backstory="You are a creative consultant helping fantasy writers refine their initial story ideas. "
                  "You specialize in identifying characters, setting, tone, conflict, and themes. "
                  "You guide the entire outlining process by establishing a clear vision.",
        llm=llm,
        allow_delegation=False,
        verbose=True,
    )
    
    worldbuilder = Agent(
        role="Worldbuilding Expert",
        goal="Design an ULTRA-DETAILED fantasy world with comprehensive systems.",
        backstory="You are a seasoned worldbuilder who constructs vivid and believable fantasy realms. "
                  "You define geography, cultures, magic systems, political structures, and mythologies "
                  "that serve as the foundation for storytelling.",
        llm=llm,
        allow_delegation=False,
        verbose=True,

    )
    
    character_architect = Agent(
        role="Character Architect",
        goal="Create detailed, multi-dimensional characters who drive the story forward.",
        backstory="You are a character development expert who crafts heroes, villains, and side characters. "
                  "You define their backstories, motivations, arcs, relationships, and emotional journeys.",
        llm=llm,
        allow_delegation=False,
        verbose=True,
       
    )
    
    plot_strategist = Agent(
        role="Plot Structure Strategist",
        goal="Generate a structured fantasy plot using traditional narrative frameworks.",
        backstory="You are a master storyteller who weaves plots using the Hero’s Journey, Three Act Structure, and other frameworks. "
                  "You ensure that the plot flows logically and engages readers from start to finish.",
        llm=llm,
        allow_delegation=False,
        verbose=True,
       
    )
    
    scene_director = Agent(
        role="Scene Breakdown Specialist",
        goal="Translate the structured plot into detailed scene-by-scene or chapter-wise outlines.",
        backstory="You specialize in breaking stories into vivid scenes. "
                  "You provide clear settings, goals, conflicts, and turning points for each scene "
                  "to help the writer visualize and write the novel more easily.",
        llm=llm,
        allow_delegation=False,
        verbose=True,
     
    )
    
    theme_advisor = Agent(
        role="Theme & Tone Advisor",
        goal="Ensure thematic consistency and emotional depth across the outline align with {user_idea}.",
        backstory="You are a literary expert who analyzes and suggests core themes, tones, and motifs. "
                  "You help the story feel emotionally resonant and stylistically consistent throughout.",
        llm=llm,
        allow_delegation=False,
        verbose=True,
      
    )
    
    consistency_checker = Agent(
        role="Continuity & Logic Checker",
        goal="Identify logical gaps, plot holes, and inconsistencies in the full outline.",
        backstory="You are an editorial specialist who reviews the full outline for quality control. "
                  "You ensure that the plot, characters, and world are coherent and nothing is left underdeveloped.",
        llm=llm,
        allow_delegation=False,
        verbose=True,
    )
    
    
    #tasks
    
    
    idea_extraction_task = Task(
    description=(
        f"1. Analyze the user's raw fantasy novel idea: {user_idea}.\n"
        "2. Identify the main characters, potential setting, central conflict, and themes.\n"
        "3. Clarify vague elements and structure the idea into a coherent story seed.\n"
        "4. Prepare the refined concept for use in worldbuilding and plotting.\n"
        "5. Generate 1000+ words with exhaustive commentary"
    ),
    expected_output="A structured 1000+ word summary of the story idea including protagonist, world hints, tone, conflict, and themes.",
    agent=idea_extractor
    )
    
    worldbuilding_task = Task(
        description=(
            "1. Use the refined idea to design a vivid and immersive fantasy world.\n"
            "2. Define the geography, magic systems, political structures, and cultural norms.\n"
            "3. Include religion, history, economic systems, and mythologies where relevant.\n"
            "4. Ensure internal consistency and creative depth across the world design.\n"
            "5.Write at least 1000 words."
        ),
        expected_output="A detailed 2000+ word world document covering terrain, magic, politics, culture, and unique lore.",
        agent=worldbuilder
    )
    
    character_task = Task(
        description=(
            "1. Create detailed character profiles for at least one protagonist, one antagonist, and two key supporting characters.\n"
            "2. Define names, roles, traits, motivations, arcs, and relationships.\n"
            "3. Ensure each character supports the core theme and plot direction.\n"
            "4. Highlight how their evolution affects the story.\n"
            "5. Write at least 800+ words."
        ),
        expected_output="Four or more character bios totaling 800+ words with backstory, arc, and inter-character dynamics.",
        agent=character_architect
    )
    
    plot_task = Task(
        description=(
            "1. Develop a plot outline using the Hero’s Journey or Three-Act Structure.\n"
            "2. Include exposition, inciting incident, rising action, climax, and resolution.\n"
            "3. Integrate major turning points, conflicts, and internal dilemmas.\n"
            "4. Align plot beats with character and theme development.\n"
            "5. Write at least 800 words."
        ),
        expected_output="A structured plot outline with 800+ words covering 8–12 major beats linked to characters and world.",
        agent=plot_strategist
    )
    
    scene_task = Task(
        description=(
            "1. Break the structured plot into a detailed scene-by-scene or chapter-wise outline.\n"
            "2. For each scene, define the setting, involved characters, scene goal, and conflict.\n"
            "3. Include pacing considerations and emotional flow.\n"
            "4. Ensure each scene advances the plot and develops theme or character.\n"
            "5. Write at least 1000+ words."
        ),
        expected_output="A comprehensive 1000+ word scene-by-scene breakdown to guide novel structure.",
        agent=scene_director
    )
     
    theme_task = Task(
        description=(
            "1. Analyze the story concept and outline to extract central themes and motifs.\n"
            "2. Identify emotional tone (e.g., epic, tragic, whimsical) and stylistic cues.\n"
            "3. Suggest recurring symbols or metaphors that unify the narrative.\n"
            "4. Ensure tonal consistency from beginning to end.\n"
            "5. Write at least 400+ words."
        ),
        expected_output="A 400+ word theme and tone guideline covering core message, style, and motifs.",
        agent=theme_advisor
    )
    
    consistency_task = Task(
        description=(
            "1. Review the complete outline for logic gaps, inconsistencies, or contradictions.\n"
            "2. Identify undeveloped or repetitive sections.\n"
            "3. Ensure character motivations, world rules, and plot developments align.\n"
            "4. Suggest direct edits or improvements for cohesion.\n"
            "5. Write at least 2000+ words."
        ),
        expected_output="A 2000+ word edited and annotated version of the full outline with feedback and fixes.",
        agent=consistency_checker
    )
    
    
    from crewai import Crew
    
    # Assemble all the tasks
    tasks = [
        idea_extraction_task,
        worldbuilding_task,
        character_task,
        plot_task,
        scene_task,
        theme_task,
        consistency_task
    ]
    
    # Define the crew with all agents and tasks
    fantasy_outline_crew = Crew(
        agents=[
            idea_extractor,
            worldbuilder,
            character_architect,
            plot_strategist,
            scene_director,
            theme_advisor,
            consistency_checker
        ],
        tasks=tasks,
        verbose=True
    )

    results=fantasy_outline_crew.kickoff(inputs={'user_idea':user_idea})
    return results.raw



def format_output(plan_text):
    # Convert plain text or Markdown to styled HTML
    html_content = markdown(plan_text)
    styled_html = f"""
    <div style='
        font-family: "Segoe UI", sans-serif;
        line-height: 1.6;
        padding: 20px;
        background-color: #ffffff;
        border-radius: 10px;
        border: 1px solid #ccc;
        box-shadow: 0 2px 10px rgba(0,0,0,0.05);
        color: #333;
    '>
        {html_content}
    </div>
    """
    return styled_html





# def create_gradio_app():
#     with gr.Blocks(css="""
#         .gradio-container {
#             background: url('https://img.freepik.com/free-ai-image/fantasy-landscape-floating-islands-illustration_355026699.htm#fromView=keyword&page=1&position=3&uuid=c5b6d86a-b71a-47da-9593-ce6dc72d2fbd&query=Fantasy+Background') no-repeat center center fixed;
#             background-size: cover;
#             font-family: 'Segoe UI', sans-serif;
#             padding: 30px;
#         }
#         .main-title {
#             text-align: center;
#             font-size: 60px;
#             font-weight: bold;
#             font-style: italic;
#             color: #FAF3E0;
#             margin-bottom: 20px;
#         }
#         .sub-title {
#             text-align: center;
#             font-size: 28px;
#             font-weight: bold;
#             font-style: italic;
#             color: #D1C6E1;
#             margin-bottom: 30px;
#         }
#         .gradio-button {
#             background-color: #A2C2E1;
#             color: white;
#             font-size: 18px;
#             padding: 14px;
#             border-radius: 8px;
#             margin-top: 20px;
#         }
#         .gradio-button:hover {
#             background-color: #A8E6CF;
#         }
#         #output-box {
#             height: 600px;
#             overflow-y: scroll;
#             padding: 20px;
#             background-color: white;
#             border: 1px solid #ccc;
#             border-radius: 12px;
#             box-shadow: 0 2px 10px rgba(0,0,0,0.05);
#         }
#     """) as demo:
#         gr.Markdown("<div class='main-title'>Fantasy Novel Outline Generator</div>")
#         gr.Markdown("<div class='sub-title'>Turn your novel idea into a rich, detailed outline with AI-powered agents.</div>")

#         with gr.Row():
#             with gr.Column(scale=1):
#                 user_idea = gr.Textbox(
#                     label="Enter Your Fantasy Novel Idea",
#                     placeholder="e.g., A young orphan discovers they're heir to a ruined magical kingdom...",
#                     lines=5,
#                     interactive=True
#                 )

#                 generate_btn = gr.Button("Generate Outline", elem_classes="gradio-button")

#             with gr.Column(scale=2):
#                 output_html = gr.HTML(label="AI-Generated Outline", elem_id="output-box")

#         # Button logic
#         generate_btn.click(
#             fn=lambda idea: format_output(generate_fantasy_outline(idea)),
#             inputs=[user_idea],
#             outputs=output_html
#         )

#     demo.launch()

import gradio as gr
import base64

# Function to encode the image
def get_base64_image(image_path):
    with open(image_path, "rb") as img_file:
        return base64.b64encode(img_file.read()).decode('utf-8')

# Your image path - REPLACE THIS WITH YOUR ACTUAL IMAGE PATH
IMAGE_PATH = "fantasy.jpeg"  # or "C:/path/to/your/background.jpg"

# Get base64 encoded image
try:
    encoded_image = get_base64_image(IMAGE_PATH)
except FileNotFoundError:
    print(f"Error: Image not found at {IMAGE_PATH}")
    encoded_image = ""

CSS = f"""
.gradio-container {{
    background: url('data:image/jpeg;base64,{encoded_image}') !important;
    background-size: cover !important;
    background-position: center !important;
    background-attachment: fixed !important;
    min-height: 100vh;
    padding: 30px;
    font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
}}

.main-title {{
    text-align: center;
    font-size: 72px !important;
    font-weight: 800;
    color: #ffffff !important;
    margin-bottom: 15px;
    text-shadow: 3px 3px 12px #000000;
    padding: 30px;
    background: rgba(20, 20, 40, 0.7);
    border-radius: 20px;
    text-transform: uppercase;
    letter-spacing: 3px;
    border: 2px solid rgba(255,255,255,0.2);
}}

.sub-title {{
    text-align: center;
    font-size: 22px;
    color: #ffffff;
    margin-bottom: 40px;
    text-shadow: 2px 2px 6px #000000;
    background: rgba(0, 0, 0, 0.6);
    padding: 15px;
    border-radius: 10px;
    max-width: 80%;
    margin-left: auto;
    margin-right: auto;
}}

.gradio-button {{
    background-color: #4CAF50 !important;
    color: white !important;
    font-size: 18px !important;
    padding: 16px 24px !important;
    border-radius: 8px !important;
    margin-top: 20px !important;
    border: none !important;
    transition: all 0.3s ease !important;
}}

.gradio-button:hover {{
    background-color: #3e8e41 !important;
    transform: scale(1.05) !important;
    box-shadow: 0 4px 12px rgba(0,0,0,0.3) !important;
}}

#output-box {{
    height: 600px;
    overflow-y: scroll;
    padding: 25px;
    background-color: rgba(255, 255, 255, 0.92) !important;
    border: 1px solid #cccccc !important;
    border-radius: 15px !important;
    box-shadow: 0 6px 20px rgba(0,0,0,0.15) !important;
}}

.textbox {{
    background: rgba(255,255,255,0.9) !important;
}}
"""



def format_output(plan_text):
    # Convert plain text or Markdown to styled HTML
    html_content = markdown(plan_text)
    styled_html = f"""
    <div style='
        font-family: "Segoe UI", sans-serif;
        line-height: 1.6;
        padding: 20px;
        background-color: #ffffff;
        border-radius: 10px;
        border: 1px solid #ccc;
        box-shadow: 0 2px 10px rgba(0,0,0,0.05);
        color: #333;
    '>
        {html_content}
    </div>
    """
    return styled_html

def create_gradio_app():
    with gr.Blocks(css=CSS) as demo:
        # gr.Markdown("""<div class='main-title'>FANTASY NOVEL OUTLINE GENERATOR</div>""")
        # gr.Markdown("""<div class='sub-title'>Transform your creative vision into a fully-developed fantasy world</div>""")

        with gr.Row():
            with gr.Column(scale=1):
                user_idea = gr.Textbox(
                    label="Enter Your Fantasy Novel Idea",
                    placeholder="A hidden prophecy... a forgotten magic... an unlikely hero...",
                    lines=5,
                    interactive=True
                )
                generate_btn = gr.Button("Craft My Outline", elem_classes="gradio-button")

            with gr.Column(scale=2):
                output_html = gr.HTML(label="Your Generated Outline",elem_id="output-box")

        generate_btn.click(
            fn=lambda idea: format_output(generate_fantasy_outline(idea)),
            inputs=[user_idea],
            outputs=output_html
        )

    return demo

if __name__ == "__main__":
    app=create_gradio_app()
    app.launch()
# import gradio as gr
# import base64

# Function to encode the image
# def get_base64_image(image_path):
#     with open(image_path, "rb") as img_file:
#         return base64.b64encode(img_file.read()).decode('utf-8')

# # Your image path - REPLACE THIS WITH YOUR ACTUAL IMAGE PATH
# IMAGE_PATH = "fantasy-landscape-floating-islands-illustration.jpg"

# # Get base64 encoded image
# try:
#     encoded_image = get_base64_image(IMAGE_PATH)
# except FileNotFoundError:
#     print(f"Error: Image not found at {IMAGE_PATH}")
#     encoded_image = ""

# CSS = f"""
# .gradio-container {{
#     background: url('data:image/jpeg;base64,{encoded_image}') !important;
#     background-size: cover !important;
#     background-position: center !important;
#     background-attachment: fixed !important;
#     min-height: 100vh;
#     padding: 30px;
#     font-family: 'Palatino Linotype', 'Book Antiqua', serif;
# }}

# /* ENHANCED MAIN TITLE */
# .main-title {{
#     text-align: center;
#     font-size: 84px !important;  /* Increased from 72px */
#     font-weight: 900;
#     color: #f8f4e6 !important;  /* Parchment color */
#     margin-bottom: 15px;
#     text-shadow: 
#         3px 3px 0 #2a2a2a,  /* Dark shadow for depth */
#         -1px -1px 0 #2a2a2a,
#         1px -1px 0 #2a2a2a,
#         -1px 1px 0 #2a2a2a,
#         1px 1px 0 #2a2a2a;
#     padding: 40px;
#     background: rgba(40, 30, 30, 0.65);  /* Deep wine background */
#     border-radius: 25px;
#     text-transform: uppercase;
#     letter-spacing: 4px;  /* More spacing between letters */
#     border: 3px solid #d4af37;  /* Gold border */
#     margin-bottom: 30px;
# }}

# /* ENHANCED SUBTITLE */
# .sub-title {{
#     text-align: center;
#     font-size: 28px !important;  /* Increased from 22px */
#     color: #f0e6d2 !important;  /* Light parchment */
#     margin-bottom: 40px;
#     text-shadow: 
#         2px 2px 4px #000000,
#         -1px -1px 0 #2a2a2a;
#     background: rgba(60, 40, 40, 0.7);  /* Darker background */
#     padding: 20px;
#     border-radius: 15px;
#     max-width: 80%;
#     margin-left: auto;
#     margin-right: auto;
#     font-style: italic;
#     border-left: 4px solid #d4af37;  /* Gold accent */
#     border-right: 4px solid #d4af37;
# }}

# .gradio-button {{
#     background: linear-gradient(to bottom, #d4af37, #b8860b) !important;
#     color: #2a2a2a !important;
#     font-size: 20px !important;
#     padding: 18px 28px !important;
#     border-radius: 10px !important;
#     margin-top: 20px !important;
#     border: 2px solid #f8f4e6 !important;
#     font-weight: bold;
#     transition: all 0.3s ease !important;
# }}

# .gradio-button:hover {{
#     background: linear-gradient(to bottom, #f8d030, #d4af37) !important;
#     transform: scale(1.05) !important;
#     box-shadow: 0 6px 16px rgba(0,0,0,0.4) !important;
# }}

# #output-box {{
#     height: 600px;
#     overflow-y: scroll;
#     padding: 30px;
#     background-color: rgba(248, 244, 230, 0.95) !important;  /* Parchment background */
#     border: 2px solid #d4af37 !important;
#     border-radius: 15px !important;
#     box-shadow: 0 8px 25px rgba(0,0,0,0.2) !important;
#     color: #333333;
# }}

# .textbox {{
#     background: rgba(255,255,255,0.95) !important;
#     border: 1px solid #d4af37 !important;
# }}
# """


# def format_output(outline):
#     return outline  # Now handled in generate_fantasy_outline

# def create_gradio_app():
#     with gr.Blocks(css=CSS) as demo:
#         # gr.Markdown("""<div class='main-title'>FANTASY NOVEL OUTLINE GENERATOR</div>""")
#         # gr.Markdown("""<div class='sub-title'>Transform your creative vision into a fully-developed fantasy world</div>""")

#         with gr.Row():
#             with gr.Column(scale=1):
#                 user_idea = gr.Textbox(
#                     label="Your Fantasy Novel Concept",
#                     placeholder="A hidden prophecy... a forgotten magic... an unlikely hero...",
#                     lines=5,
#                     interactive=True
#                 )
#                 generate_btn = gr.Button("Craft My Outline", elem_classes="gradio-button")

#             with gr.Column(scale=2):
#                 output_html = gr.HTML(label="Your Generated Outline", elem_id="output-box")
#         output = gr.HTML(label="Your Outline") 
#         generate_btn.click(
#             fn=lambda idea: generate_fantasy_outline(idea),
#             inputs=[user_idea],
#             outputs=output
#         )

#     return demo

# if __name__ == "__main__":
#     app = create_gradio_app()
#     app.launch()