Update app.py

app.py

@@ -6,6 +6,7 @@ import html
 import os
 import logging
 import subprocess
+from urllib.parse import quote
 
 # Set up logging
 logging.basicConfig(level=logging.INFO)
@@ -489,7 +490,7 @@ def generate_portfolio(file, theme="Light"):
     try:
         if not file:
             logger.error("No file uploaded.")
-            return "Error: No file uploaded. Please upload a resume file (PDF or text).", "Error: No file uploaded. Please upload a resume file (PDF or text)."
+            return "Error: No file uploaded. Please upload a resume file (PDF or text).", "Error: No file uploaded. Please upload a resume file (PDF or text).", None
         
         # In Hugging Face Spaces, file.name is the path to the uploaded file
         file_path = file.name if hasattr(file, 'name') else file
@@ -503,7 +504,7 @@ def generate_portfolio(file, theme="Light"):
         
         if "Error" in resume_text:
             logger.error(f"Failed to extract text: {resume_text}")
-            return resume_text, resume_text
+            return resume_text, resume_text, None
         
         # Extract details from the resume
         name, email, phone, education_html, experience_html, projects_html, skills_html, involvement_html, skills_summary, skills = extract_details(resume_text)
@@ -525,33 +526,17 @@ def generate_portfolio(file, theme="Light"):
         with open("index.html", "w", encoding="utf-8") as f:
             f.write(html_code)
         
-        # Preview HTML for Gradio (simplified, as Gradio can't render full animations)
+        # Display the full website in an iframe
         html_preview = f"""
-        <div style='font-family: Arial; max-width: 800px; margin: auto; padding: 20px; background: #fff; box-shadow: 0 0 10px rgba(0,0,0,0.1);'>
-            <h1 style='text-align: center;'>{name}</h1>
-            <p style='text-align: center;'>{skills_summary}</p>
-            <h2>Education</h2>
-            {education_html}
-            <h2>Work Experience</h2>
-            {experience_html}
-            <h2>Projects</h2>
-            {projects_html}
-            <h2>Skills</h2>
-            <div style='display: flex; flex-wrap: wrap; gap: 10px;'>
-                {skills_html}
-            </div>
-            <h2>Involvement</h2>
-            {involvement_html}
-            <h2>Contact</h2>
-            <p>Email: {email} | Phone: {phone}</p>
-        </div>
+        <iframe src="/file=index.html" width="100%" height="600px" style="border:none;"></iframe>
+        <p style="text-align: center;"><a href="/file=index.html" target="_blank">View Full Portfolio in New Tab</a></p>
         """
         
         logger.info("Successfully generated portfolio.")
-        return html_preview, html_code
+        return html_preview, html_code, None
     except Exception as e:
         logger.error(f"Error generating portfolio: {str(e)}")
-        return f"Error generating portfolio: {str(e)}", f"Error generating portfolio: {str(e)}"
+        return f"Error generating portfolio: {str(e)}", f"Error generating portfolio: {str(e)}", None
 
 # Gradio Interface
 with gr.Blocks() as interface:
@@ -562,11 +547,12 @@ with gr.Blocks() as interface:
     
     html_output = gr.HTML(label="Portfolio Preview")
     code_output = gr.Textbox(label="Downloadable HTML/CSS Code", lines=10)
+    error_output = gr.Textbox(label="Error (if any)", visible=False)
     
     generate_btn.click(
         fn=generate_portfolio,
         inputs=[file_input, theme_input],
-        outputs=[html_output, code_output]
+        outputs=[html_output, code_output, error_output]
     )
 
 interface.launch()