Update app.py

app.py

@@ -1,4 +1,7 @@
+
 import os
+from pathlib import Path
+from google.colab import userdata
 import litellm
 from crewai import Agent, Task, Crew, Process
 from crewai_tools import SerperDevTool
@@ -6,12 +9,14 @@ import pdfplumber
 from docx import Document
 import gradio as gr
 
+
+
 # Set up API keys
 litellm.api_key = "AIzaSyAlNlvB2unkbmkL6zweKVkjYEOVqIgovw4"
 os.environ['SERPER_API_KEY'] = "d9295b6ec268aad1fa375484f1e3ce556329973d"
 
 # Define the LLM
-llm = "gemini/gemini-1.5-flash-exp-0827"  # Your LLM model
+llm = "gemini/gemini-1.5-flash-exp-0827" 
 
 # Initialize the tool for internet searching capabilities
 tool = SerperDevTool()
@@ -23,10 +28,11 @@ cv_analysis_agent = Agent(
     verbose=True,
     memory=True,
     backstory=(
-        "As a CV Analyzer, you are skilled in identifying key information "
-        "from resumes to aid in building effective portfolios."
-        "You can add relevant skills and job responsibilities evaluating the whole cv."
-    ),
+    "You are an expert CV Analyzer with a keen eye for detail. Your role is to meticulously examine the provided CV, "
+    "identifying and extracting key skills, experiences, accomplishments, and areas for improvement. "
+    "Your analysis should highlight strengths and suggest enhancements that would make the portfolio more competitive."
+),
+
     tools=[tool],
     llm=llm,
     allow_delegation=True
@@ -35,12 +41,14 @@ cv_analysis_agent = Agent(
 # Create the Portfolio Generation Agent
 portfolio_generation_agent = Agent(
     role='Portfolio Generator',
-    goal='Generate a beautiful static HTML/CSS/JS landing portfolio webpage based on CV analysis.',
+    goal='Generate a professional HTML/CSS/JS responsive landing portfolio webpage based on {cv} analysis.',
     verbose=True,
     memory=True,
     backstory=(
-        "As a Portfolio Generator, you craft engaging web pages with effective functionalities and color combinations "
-        "to showcase individual talents and experiences with the best user experience."
+    "As a Responsive Portfolio Generator, your expertise lies in creating visually appealing and user-friendly web pages. "
+    "Based on the CV analysis, you will generate a professional HTML/CSS/JS portfolio. "
+    "Ensure the design reflects the individual's strengths and experiences while incorporating effective functionality. "
+    "Consider responsiveness, color schemes, and navigation for an optimal user experience."
     ),
     tools=[tool],
     llm=llm,
@@ -50,29 +58,28 @@ portfolio_generation_agent = Agent(
 # Research task for CV analysis
 cv_analysis_task = Task(
     description=(
-        "Analyze the provided {cv} and identify key skills, experiences, "
-        "and accomplishments. Highlight notable projects and educational background."
+        "Analyze the provided {cv} in detail. Identify and summarize key skills, experiences, and notable accomplishments. "
+        "Highlight educational background and suggest potential enhancements to improve the overall presentation and competitiveness of the CV."
     ),
-    expected_output='A summary of skills, experiences, and projects formatted for a portfolio.',
+    expected_output='A detailed summary of skills, experiences, accomplishments, and improvement suggestions formatted for a portfolio.',
     tools=[tool],
     agent=cv_analysis_agent,
 )
 
+
 # Writing task for portfolio generation with enhanced UI requirements
+
 portfolio_task = Task(
     description=(
-        "Generate a static HTML/CSS/JS landing portfolio with a name as header in top, navbar for different sections, beautiful and responsive design. "
-        "Ensure that the layout is clean, with sections for skills, projects, experiences, certifications, publications, and contact details if present in the CV. "
-        "Include a footer that does not overlap with the content. "
-        "Use a modern color palette and incorporate CSS frameworks if necessary, "
-        "but provide everything embedded in the HTML file. "
-        "The output should be a complete HTML document starting from <html> to </html>, ready to deploy."
+        "Generate a responsive HTML/CSS portfolio webpage based on the given CV analysis. "
+        "Include a navbar with the individual's name, and sections for skills, projects, experiences, certifications, and contact information. "
+        "Ensure the layout is clean and visually appealing with a light/dark theme toggle option. "
+        "Embed CSS/JS directly into the HTML for easy deployment, and optimize for both desktop and mobile viewing."
     ),
-    expected_output='A complete HTML/CSS/JS code content only for a portfolio website in a single .html file',
+    expected_output='A complete and responsive HTML document ready for deployment, showcasing the individual’s strengths.',
     tools=[tool],
     agent=portfolio_generation_agent,
-    async_execution=False,
-    output_file='portfolio.html'  # Output as HTML file
+    async_execution=True,
 )
 
 # Function to read CV from PDF or DOCX file
@@ -100,8 +107,7 @@ crew = Crew(
     process=Process.sequential,
 )
 
-# Function to process CV and generate portfolio
-import re
+
 
 # Function to process CV and generate portfolio
 def process_cv(file):
@@ -116,28 +122,50 @@ def process_cv(file):
         html_output = str(result)
 
         # Use replace to remove '''html''' and ''' from the output
-        clean_html_output = html_output.replace("'''html'''", '').replace("'''", '').strip()
+        clean_html_output = html_output.replace("```html", '').replace("```", '').strip()
 
         return clean_html_output  # Return the cleaned HTML
     except Exception as e:
         return f"Error: {e}"
 
-# Gradio UI using Blocks
-with gr.Blocks() as iface:
-    gr.Markdown("# CV to AI Enhanced Portfolio Webpage Generator")
-    gr.Markdown("Upload your CV in PDF or DOCX format to analyze its content and generate a HTML/CSS/JS code for your portfolio webpage.")
-    
-    # File input for uploading CV
-    cv_input = gr.File(label="Upload your CV (.pdf or .docx)")
+
+
+
+def save_html_to_file(html_content):
+    output_file_path = "Portfolio_generated_by_FiftyBit.html"
+    with open(output_file_path, "w") as f:
+        f.write(html_content)
+    return output_file_path
+
+def upload_file(filepath):
+    name = Path(filepath).name
+    html_content = process_cv(filepath)  # Get HTML content from the CV
+
+    clean_html_output = html_content.replace("```html", '').replace("```", '').strip()
+    gr.Markdown("# Preview of Your Portfolio:") 
+    # Save the HTML content to a file
+    file_path = save_html_to_file(clean_html_output)
     
-    # Output textbox for generated HTML
-    output_textbox = gr.Textbox(label="Generated HTML", lines=20, placeholder="Your generated HTML/CSS/JS code will appear here...", interactive=True)
+    return clean_html_output, gr.UploadButton(visible=False), gr.DownloadButton(label=f"Download Code", value=file_path, visible=True)
+
+def download_file():
+    return [gr.UploadButton(label=f"Regenerate", visible=True), gr.DownloadButton(visible=False)]
+
+with gr.Blocks() as demo:
+    gr.Markdown("# CV-2-Portfolio Site Generator")
+    gr.Markdown("Upload your CV in PDF or DOCX format for analysis and portfolio webpage generation.")
     
-    # Process button
-    process_button = gr.Button("Generate Portfolio")
+  
+    u = gr.UploadButton("Upload CV (.pdf or .docx)", file_count="single")
+    d = gr.DownloadButton("Download Portfolio", visible=False)
     
-    # Define the button actions
-    process_button.click(fn=process_cv, inputs=cv_input, outputs=output_textbox)
+    output_preview = gr.HTML(label="Webpage Preview"  , value="<style>iframe { width: 100%; height: 600px; }</style><iframe srcdoc='Portfolio_generated_by_FiftyBit.html'></iframe></style>"  )  # HTML output for preview
 
-# Launch the Gradio interface
-iface.launch()
+              
+
+    # Connect the upload button to the upload_file function
+    u.upload(upload_file, u, [output_preview, u, d])
+    
+    # Handle download button click
+    d.click(download_file, None, [u, d])
+demo.launch(debug=True)