Update app.py

app.py

@@ -135,29 +135,31 @@ def save_html_to_file(html_content):
     return output_file_path
 
 
+import html
+
 def upload_file(filepath):
     name = Path(filepath).name
     html_content = process_cv(filepath)  # Get HTML content from the CV
 
-    # Clean the HTML content (but keep this simple to start with)
+    # Clean the HTML content and escape it for proper iframe embedding
     clean_html_output = html_content.replace("```html", '').replace("```", '').strip()
+    escaped_html_content = html.escape(clean_html_output)  # Escape HTML content
 
-    # Debugging print to check the HTML content
-    print("HTML content:", clean_html_output)
+    # Debugging print to check the escaped HTML content
+    #print("Escaped HTML content:", escaped_html_content)
 
     # Save the cleaned HTML content to a file (if you still want this feature)
     file_path = save_html_to_file(clean_html_output)
 
     # Return a full HTML string with embedded iframe for preview
     iframe_html = f"""
-    <iframe srcdoc="{clean_html_output}" style="width:100%; height:1000px; border:none; overflow:auto;"></iframe>
+    <iframe srcdoc="{escaped_html_content}" style="width:100%; height:1000px; border:none; overflow:auto;"></iframe>
     """
     return iframe_html, 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)]
 
-
 # Gradio App
 with gr.Blocks() as demo:
     gr.Markdown("<center><h1> CV-2-Portfolio Site Generator</center></h1>")