Update app.py

app.py

@@ -1,10 +1,9 @@
-# app.py
 import gradio as gr
 import json
 import os
 import random
 import string
-from html_generator import generate_website_html
+from html_generator import generate_website_html, generate_navbar, generate_hero_section, generate_features_section, generate_pricing_section, generate_testimonials_section, generate_contact_section
 
 # CSS for styling the interface
 css = """
@@ -71,17 +70,26 @@ example_prompts = [
 # Function to handle website generation
 def create_website(prompt):
     site_id = generate_site_id()
-    
+
     try:
         # Generate HTML content based on the prompt
-        html_content = generate_website_html(prompt)
-        
+        title = "Generated Website"
+        sections = [
+            generate_navbar(title),
+            generate_hero_section("Welcome to Your Website", "This is a demo website generated from your prompt."),
+            generate_features_section(),
+            generate_pricing_section(),
+            generate_testimonials_section(),
+            generate_contact_section()
+        ]
+        html_content = generate_website_html(title, sections)
+
         # Save the HTML content to a file
         os.makedirs('websites', exist_ok=True)
         file_path = f"websites/{site_id}.html"
         with open(file_path, "w") as f:
             f.write(html_content)
-        
+
         # In a real Hugging Face Space, you might use a different mechanism to serve the file
         # For demo purposes, we'll return the HTML directly
         preview_html = f"""
@@ -93,9 +101,9 @@ def create_website(prompt):
             <iframe srcdoc="{html_content.replace('"', '&quot;')}" width="100%" height="600px"></iframe>
         </div>
         """
-        
+
         return preview_html, html_content
-    
+
     except Exception as e:
         return f"Error generating website: {str(e)}", ""
 
@@ -108,38 +116,38 @@ with gr.Blocks(css=css) as demo:
             <p>Generate a complete website from a simple prompt. Describe what you want, and we'll create it.</p>
         </div>
         """)
-        
+
         with gr.Row():
             prompt_input = gr.Textbox(
                 placeholder="Describe the website you want to create...",
                 label="Your Website Prompt",
                 elem_classes="prompt-box"
             )
-        
+
         with gr.Row():
             generate_btn = gr.Button("Generate Website", elem_classes="generate-btn")
-        
+
         with gr.Row():
             gr.HTML("<p>Need inspiration? Try one of these examples:</p>")
-        
+
         with gr.Row():
             example_buttons = [gr.Button(f"Example {i+1}", elem_classes="example-btn") for i in range(5)]
-        
+
         with gr.Accordion("Advanced Options", open=False):
             gr.Checkbox(label="Include responsive design", value=True)
             gr.Checkbox(label="Generate CSS animations", value=False)
             gr.Dropdown(["Light", "Dark", "Colorful"], label="Color Theme", value="Light")
-        
+
         output_preview = gr.HTML(elem_classes="output-container")
         output_code = gr.Code(language="html", label="HTML Code", visible=False)
-        
+
         # Set up button actions
         generate_btn.click(create_website, inputs=[prompt_input], outputs=[output_preview, output_code])
-        
+
         # Set up example button actions
         for i, btn in enumerate(example_buttons):
             btn.click(lambda _, i=i: example_prompts[i], outputs=prompt_input)
 
 # Launch the app
 if __name__ == "__main__":
-    demo.launch(share=True)
+    demo.launch(share=True)