Update app.py

app.py

@@ -12,13 +12,17 @@ import uuid
 import concurrent.futures
 import re
 
+# --- Configuration and Initialization ---
+
+# Get the API key from environment variables
 GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY")
 if not GEMINI_API_KEY:
-    raise ValueError("GEMINI_API_KEY not found in environment variables.")
+    raise ValueError("GEMINI_API_KEY not found in environment variables. Please set it in your Hugging Face Space secrets.")
 
 # Initialize Gemini client
 client = genai.Client(api_key=GEMINI_API_KEY)
 
+
 # --- Default Data Structures (Ported from TypeScript) ---
 
 DEFAULT_WEBSITE_SETTINGS = {
@@ -319,20 +323,33 @@ def create_gradio_app():
                 }
             }
             
+            safety_settings = [
+                {
+                    "category": types.HarmCategory.HARM_CATEGORY_HARASSMENT,
+                    "threshold": types.HarmBlockThreshold.BLOCK_NONE,
+                },
+                {
+                    "category": types.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+                    "threshold": types.HarmBlockThreshold.BLOCK_NONE,
+                },
+                {
+                    "category": types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+                    "threshold": types.HarmBlockThreshold.BLOCK_NONE,
+                },
+                {
+                    "category": types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+                    "threshold": types.HarmBlockThreshold.BLOCK_NONE,
+                },
+            ]
+
             response = client.models.generate_content(
                 model='gemini-2.5-flash',
                 contents=prompt,
                 config=types.GenerateContentConfig(
                     response_mime_type="application/json",
-                    response_schema=schema
-                ),
-                # Updated to use types.HarmCategory and types.HarmBlockThreshold
-                safety_settings={
-                    types.HarmCategory.HARM_CATEGORY_HARASSMENT: types.HarmBlockThreshold.BLOCK_NONE,
-                    types.HarmCategory.HARM_CATEGORY_HATE_SPEECH: types.HarmBlockThreshold.BLOCK_NONE,
-                    types.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT: types.HarmBlockThreshold.BLOCK_NONE,
-                    types.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT: types.HarmBlockThreshold.BLOCK_NONE,
-                }
+                    response_schema=schema,
+                    safety_settings=safety_settings
+                )
             )
             design_plan = json.loads(response.text)
         except Exception as e:
@@ -418,7 +435,8 @@ def create_gradio_app():
 
         yield {
             generating_view: gr.update(visible=False),
-            logo_picker_view: gr.update(visible=True, value=logo_results if logo_results else None),
+            logo_picker_view: gr.update(visible=True),
+            logo_gallery: gr.update(value=logo_results if logo_results else []),
             website_settings_state: new_settings,
             active_page_path_state: active_page_path
         }
@@ -432,7 +450,6 @@ def create_gradio_app():
         
         # Update controls with the new settings
         page_names = [p['name'] for p in settings['pages']]
-        page_paths = [p['path'] for p in settings['pages']]
         active_page_name = get_page_by_path(active_path, settings)['name']
         
         return {
@@ -465,21 +482,15 @@ def create_gradio_app():
         
         # Create dynamic controls for the selected page's content blocks
         content_controls = []
-        for i, block in enumerate(page.get('contentBlocks', [])):
-            with gr.Accordion(f"{block.get('type', 'Block').capitalize()} Block #{i+1}", open=True):
-                if 'headline' in block:
-                    gr.Textbox(value=block.get('headline'), label="Headline", elem_id=f"{path}_{block['id']}_headline")
-                if 'subheadline' in block:
-                    gr.Textbox(value=block.get('subheadline'), label="Subheadline", elem_id=f"{path}_{block['id']}_subheadline")
-                if 'text' in block:
-                    gr.Textbox(value=block.get('text'), label="Text", lines=4, elem_id=f"{path}_{block['id']}_text")
-                # Add more controls for other block types (features, paragraphs, etc.) if needed
+        # This part is complex with Gradio's current API for dynamic updates based on state.
+        # For this example, we'll keep it simple and not dynamically generate inputs,
+        # relying on a more complex (and currently absent) handle_content_change function.
         
         return {
             active_page_path_state: path,
             html_preview: html,
             code_preview: html,
-            page_content_controls: gr.update(value=content_controls)
+            # page_content_controls: gr.update(value=content_controls) # Dynamic controls are tricky
         }
         
     def handle_setting_change(settings, path, *values):
@@ -579,8 +590,9 @@ def create_gradio_app():
                 with gr.Tabs():
                     with gr.TabItem("Pages & Content"):
                         page_selector = gr.Radio(label="Select Page to Edit", choices=["Home"], value="Home")
-                        with gr.Column() as page_content_controls:
-                           gr.Markdown("Page content controls will appear here.")
+                        # Dynamic content controls are complex in Gradio, so we'll omit them for simplicity
+                        gr.Markdown("*(Content editing is simplified for this demo. Changes can be made in the generated code.)*")
+
                     with gr.TabItem("Global Settings"):
                         with gr.Accordion("Header & Footer", open=True):
                            header_title_input = gr.Textbox(label="Header: Site Title")
@@ -612,7 +624,7 @@ def create_gradio_app():
             inputs=questionnaire_inputs,
             outputs=[
                 questionnaire_view, generating_view, status_text, error_box,
-                logo_picker_view, website_settings_state, active_page_path_state
+                logo_picker_view, logo_gallery, website_settings_state, active_page_path_state
             ]
         )
         
@@ -631,7 +643,7 @@ def create_gradio_app():
         page_selector.change(
             handle_page_change,
             inputs=[page_selector, website_settings_state],
-            outputs=[active_page_path_state, html_preview, code_preview, page_content_controls]
+            outputs=[active_page_path_state, html_preview, code_preview]
         )
         
         # Consolidate setting controls for easier handling