Update app.py

app.py

@@ -15,7 +15,7 @@ litellm.api_key = "AIzaSyAlNlvB2unkbmkL6zweKVkjYEOVqIgovw4"
 os.environ['SERPER_API_KEY'] = "d9295b6ec268aad1fa375484f1e3ce556329973d"
 
 # Define the LLM
-llm = "gemini/gemini-1.5-flash-exp-0827" 
+llm = "gemini/gemini-1.5-flash-exp-0827"  # Your LLM model
 
 # Initialize the tool for internet searching capabilities
 tool = SerperDevTool()
@@ -129,42 +129,55 @@ def process_cv(file):
 
 
 
-
 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 the HTML content (but keep this simple to start with)
     clean_html_output = html_content.replace("```html", '').replace("```", '').strip()
-    gr.Markdown("# Preview of Your Portfolio:") 
-    # Save the HTML content to a file
+
+    # Debugging print to check the HTML content
+    print("HTML content:", clean_html_output)
+
+    # Prepare an iframe to embed the HTML content
+    iframe_html = f"<iframe srcdoc='{clean_html_output}' style='width:100%; height:600px; border:none;'></iframe>"
+
+    # Debugging print to check the iframe HTML
+    # print("Iframe HTML:", iframe_html)
+
+    # Save the cleaned HTML content to a file (if you still want this feature)
     file_path = save_html_to_file(clean_html_output)
-    
-    return clean_html_output, gr.UploadButton(visible=False), gr.DownloadButton(label=f"Download Code", value=file_path, visible=True)
+
+    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)]
 
+
 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.")
-    
-  
+    gr.Markdown("<center><h1> CV-2-Portfolio Site Generator</center></h1>")
+    gr.Markdown("<center><h2>Upload your CV in PDF or DOCX format for analysis and portfolio webpage generation.</center></h1>")
+
     u = gr.UploadButton("Upload CV (.pdf or .docx)", file_count="single")
     d = gr.DownloadButton("Download Portfolio", visible=False)
-    
-    output_preview = gr.HTML(label="Webpage Preview"  , value="<style>iframe { width: 100%; height: 600px; }</style><iframe srcdoc='Portfolio_generated_by_FiftyBit.html'></iframe>"  )  # HTML output for preview
 
-              
+    # HTML output preview section (which will be updated with iframe content after file processing)
+    output_preview = gr.HTML(value="<div style='width:100%; height:600px; border:1px solid #ccc; text-align:center;'>Upload a file to preview the generated portfolio</div>")
 
-    # Connect the upload button to the upload_file function
+    # Connect the upload button to the upload_file function and update the output preview
     u.upload(upload_file, u, [output_preview, u, d])
-    
+
     # Handle download button click
     d.click(download_file, None, [u, d])
-demo.launch(debug=True)
+
+demo.launch(debug=True)
+
+
+