diff --git a/app.py b/app.py
index 4930520..21cbfba 100644
--- a/app.py
+++ b/app.py
@@ -51,12 +51,6 @@ def format_response(response):
     formatted_response = "<ul>" + "\n".join(formatted_lines) + "</ul>" if formatted_lines else response
     return formatted_response.replace("<ul></ul>", "")  # Remove empty list tags
 
-def calculate_progress(sections):
-    completed_sections = [section for section in sections if section['content']]
-    total_sections = 4  # description, business_model, setup_process, budget
-    progress = int((len(completed_sections) / total_sections) * 100)
-    return progress
-
 
 # adding visualiztions
 
@@ -145,88 +139,92 @@ def pedict():
     session.clear()  # Clear session at the start
     return render_template('predict.html')
 
-@app.route('/download', methods=['POST'])
-def download():
-    service_name = request.form['service_name']
-    approved_sections = session.get('approved_sections', {})
-    # You need to implement the logic to convert approved_sections into a downloadable format
-    # For example:
-    download_content = "Your Agrotourism Service Plan\n"
-    for section_name, section_content in approved_sections.items():
-        download_content += f"{section_name.title()}\n{section_content}\n\n"
-    
-    # Create a response with the content
-    response = make_response(download_content)
-    response.headers["Content-Disposition"] = "attachment; filename=service_plan.txt"
-    response.headers["Content-Type"] = "text/plain"
-    return response
 
 @app.route('/generate', methods=['GET', 'POST'])
 def generate():
+    # Initialize or get current section and form data from session
     if request.method == 'POST':
-        service_name = session.get('service_name', request.form['service_name'])
-        session['service_name'] = service_name
-
+        # Ensure form_data is updated or initialized correctly
+        form_data = {
+            'service_name': request.form['service_name'],
+            'land_size': request.form.get('land_size', 'N/A'),
+            'biodiversity': request.form.get('biodiversity', 'N/A'),
+            'budget': request.form.get('budget', 'N/A'),
+            'infrastructure': request.form.getlist('infrastructure[]'),
+        }
+        session['form_data'] = form_data
+
+        # Initialize or get sections from session
         sections = session.get('sections', [])
-        approved_sections = session.get('approved_sections', {})
-        current_section = request.form.get('section', 'description')
-        is_final_section = False
+        if sections:
+            # Determine the current section from the last entry in sections
+            current_section = sections[-1]['name']
+        else:
+            # Default to starting with 'description'
+            current_section = 'description'
+
+        # Determine next section
+        section_order = ['description', 'business_model', 'setup_process', 'budget']
+        current_index = section_order.index(current_section)
 
         if 'accept' in request.form:
-            # Store the current approved section content
-            approved_sections[current_section] = sections[-1]['content']
-            session['approved_sections'] = approved_sections
-
-            next_section_map = {
-                'description': 'business_model',
-                'business_model': 'setup_process',
-                'setup_process': 'budget',
-                'budget': 'end'
-            }
-            next_section = next_section_map.get(current_section)
-
-            if next_section == 'end':
-                is_final_section = True
-                # Prepare for download, show download button
-                progress = 100  # Final section implies 100% progress
-                # Render the template with the final section and show the download button
-                return render_template('generate.html', sections=sections, service_name=service_name, 
-                                       is_final_section=is_final_section, progress=progress, 
-                                       approved_sections=approved_sections, show_download=True)
+            # If accepting current section, move to next unless at the end
+            if current_index + 1 < len(section_order):
+                current_section = section_order[current_index + 1]
             else:
-                # Append the new section if it's not the end
-                sections.append({'name': next_section, 'content': ''})
-                current_section = next_section
-
+                # If at the last section, perhaps redirect or indicate completion
+                return render_template('complete.html', sections=sections, service_name=form_data['service_name'])
         elif 'regenerate' in request.form:
-            # If regenerate is clicked, simply re-display the current section with the old content
-            # There's no need to append a new section or generate new content
-            pass
-
+            # If regenerating, stay on current_section
+            pass  # current_section remains the same
         else:
-            # New section is being started without any user approval action
-            sections.append({'name': current_section, 'content': ''})
+            # Handling for generating the first section or when not accepting/regenerating
+            if not sections:  # If starting fresh
+                current_section = 'description'
 
-        # Generate content for the current section
-        prompt_parts = [f"Generate {current_section} for {service_name} in the context of agrotourism."]
-        response = generate_description(prompt_parts, get_generation_config(), get_safety_settings())
+        # Generate content for the current or next section
+        prompt_template = create_prompt_template(current_section, form_data)
+        response = generate_description(prompt_template, get_generation_config(), get_safety_settings())
         formatted_response = format_response(response)
-        sections[-1]['content'] = formatted_response
 
+        # Update sections list appropriately
+        if 'accept' in request.form and sections:
+            # Replace last section content if regenerating; otherwise, append new section
+            sections[-1] = {'name': current_section, 'content': formatted_response}
+        else:
+            sections.append({'name': current_section, 'content': formatted_response})
+
+        # Save updated sections and form_data back to session
         session['sections'] = sections
-        progress = calculate_progress(sections)
 
-        return render_template('generate.html', sections=sections, service_name=service_name, 
-                               is_final_section=is_final_section, progress=progress, 
-                               approved_sections=approved_sections, show_download=False)
+        # Calculate progress for UI feedback
+        progress = calculate_progress(sections, section_order)
+
+        # Render the template with updated context
+        return render_template('generate.html', sections=sections, current_section=current_section, service_name=form_data['service_name'], progress=progress)
     else:
-        # Clear the session for a new start
-        session.pop('sections', None)
-        session.pop('approved_sections', None)
-        return render_template('generate.html', sections=[], service_name=None, 
-                               is_final_section=False, progress=0, approved_sections={}, 
-                               show_download=False)
+        # Clear the session for a new start and render the initial form
+        session.clear()
+        return render_template('generate.html', sections=[], current_section='description', service_name='', progress=0)
+
+def calculate_progress(sections, section_order):
+    # Calculate the progress based on sections completed
+    completed_sections = len(sections)
+    total_sections = len(section_order)
+    progress = int((completed_sections / total_sections) * 100)
+    return progress
 
+def create_prompt_template(current_section, form_data):
+    # Create a detailed prompt for the current section based on form_data
+    # Adjust this function as necessary to tailor prompts for each section
+    prompt_parts = [
+        f"Generate {current_section} for {form_data['service_name']}",
+        f"Land Size: {form_data['land_size']} hectares",
+        f"Biodiversity: {form_data['biodiversity']}",
+        f"Budget: INR {form_data['budget']}",
+        f"Existing Infrastructure: {', '.join(form_data['infrastructure'])}.",
+    ]
+    return " ".join(prompt_parts)
 
 
 if __name__ == '__main__':
diff --git a/sampleapp b/sampleapp
new file mode 100644
index 0000000..0f347c5
--- /dev/null
+++ b/sampleapp
@@ -0,0 +1,251 @@
+from flask import Flask, render_template, request, session
+import google.generativeai as genai
+import os
+# import torch
+# from PIL import Image
+# import numpy as np
+# from flask import jsonify, send_from_directory
+# from werkzeug.utils import secure_filename
+
+
+app = Flask(__name__)
+app.secret_key = os.urandom(24)  # Necessary for session management
+
+API_KEY = os.getenv('GENAI_API_KEY')
+genai.configure(api_key=API_KEY)
+
+def generate_description(prompt_parts, generation_config, safety_settings):
+    model = genai.GenerativeModel(model_name="gemini-pro",
+                                  generation_config=generation_config,
+                                  safety_settings=safety_settings)
+    response = model.generate_content(prompt_parts)
+    clean_response = response.text.replace("**", "")
+    return clean_response
+
+def get_generation_config():
+    return {
+        "temperature": 0.9,
+        "top_p": 1,
+        "top_k": 1,
+        "max_output_tokens": 2048,
+    }
+
+def get_safety_settings():
+    return [
+        {"category": "HARM_CATEGORY_HARASSMENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+        {"category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+        {"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+        {"category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "BLOCK_MEDIUM_AND_ABOVE"},
+    ]
+
+def format_response(response):
+    # Convert markdown bullet points and bold text to HTML
+    response = response.replace("**", "<strong>").replace("<strong>", "</strong>", 1)  # Bold
+    lines = response.split('\n')
+    formatted_lines = []
+    for line in lines:
+        if line.startswith('- ') or line.isdigit():  # Bullet points or numerical points
+            formatted_lines.append(f"<li>{line[2:] if line.startswith('- ') else line}</li>")
+        else:
+            formatted_lines.append(line)
+    formatted_response = "<ul>" + "\n".join(formatted_lines) + "</ul>" if formatted_lines else response
+    return formatted_response.replace("<ul></ul>", "")  # Remove empty list tags
+
+
+def calculate_progress(sections, section_order):
+    # Calculate progress based on the number of completed sections
+    progress = int((len(sections) / len(section_order)) * 100)
+    return progress
+
+
+# adding visualiztions
+
+# device = "cuda" if torch.cuda.is_available() else "cpu"
+
+# UPLOAD_FOLDER = 'uploads'
+# GENERATED_FOLDER = 'generated'
+# app.config['UPLOAD_FOLDER'] = UPLOAD_FOLDER
+# app.config['GENERATED_FOLDER'] = GENERATED_FOLDER
+
+# os.makedirs(UPLOAD_FOLDER, exist_ok=True)
+# os.makedirs(GENERATED_FOLDER, exist_ok=True)
+
+# def generate_image_with_ml_model(image_path, prompt):
+#     # Load the image
+#     input_image = load_image(image_path).to(device)
+
+#     # Initialize the depth estimator
+#     depth_estimator = pipeline("depth-estimation", device=device)
+    
+#     # Process to obtain depth map
+#     depth_map = get_depth_map(input_image, depth_estimator)  # Assuming get_depth_map is defined similarly to your Colab code
+
+#     # Initialize the ControlNet model and pipeline
+#     controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-normal", torch_dtype=torch.float16, use_safetensors=True).to(device)
+#     pipe = StableDiffusionControlNetImg2ImgPipeline.from_pretrained(
+#         "runwayml/stable-diffusion-v1-5",
+#         controlnet=controlnet,
+#         torch_dtype=torch.float16,
+#         use_safetensors=True
+#     ).to(device)
+#     pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
+#     pipe.enable_model_cpu_offload()
+
+#     # Generate the image
+#     output = pipe(prompt=prompt, image=input_image, control_image=depth_map).images[0]
+
+#     # Convert tensor to PIL Image for saving
+#     output_image = Image.fromarray(output.mul(255).clamp(0, 255).byte().cpu().numpy().astype(np.uint8).transpose(1, 2, 0))
+    
+#     return output_image
+
+# @app.route('/generate-image', methods=['POST'])
+# def generate_image_endpoint():
+#     if 'image' not in request.files:
+#         return jsonify({'error': 'No image part'}), 400
+#     file = request.files['image']
+#     prompt = request.form.get('prompt', '')  # Get the prompt from the form data
+#     if file.filename == '':
+#         return jsonify({'error': 'No selected file'}), 400
+#     if file and prompt:
+#         filename = secure_filename(file.filename)
+#         input_filepath = os.path.join(app.config['UPLOAD_FOLDER'], filename)
+#         file.save(input_filepath)
+        
+#         # Generate the image
+#         output_image = generate_image_with_ml_model(input_filepath, prompt)
+#         output_filename = f"generated_{filename}"
+#         output_filepath = os.path.join(app.config['GENERATED_FOLDER'], output_filename)
+#         output_image.save(output_filepath)
+        
+#         return jsonify({'generatedImageUrl': f'/generated/{output_filename}'})
+#     else:
+#         return jsonify({'error': 'Invalid request'}), 400
+
+# @app.route('/generated/<filename>')
+# def generated_image(filename):
+#     return send_from_directory(app.config['GENERATED_FOLDER'], filename)
+
+
+@app.route('/visualize')
+def visualize():
+    return render_template('visualize.html')
+
+@app.route('/docs')
+def docs():
+    return render_template('docs.html')
+
+@app.route('/')
+def index():
+    session.clear()  # Clear session at the start
+    return render_template('index.html')
+
+@app.route('/predict')
+def pedict():
+    session.clear()  # Clear session at the start
+    return render_template('predict.html')
+
+
+@app.route('/generate', methods=['GET', 'POST'])
+def generate():
+    current_section = 'description'
+
+    if request.method == 'POST':
+
+        if 'service_name' in request.form:
+            session['form_data'] = {
+                'service_name': request.form['service_name'],
+                'land_size': request.form.get('land_size', 'N/A'),
+                'biodiversity': request.form.get('biodiversity', 'N/A'),
+                'budget': request.form.get('budget', 'N/A'),
+                'infrastructure': request.form.getlist('infrastructure[]'),
+            }
+
+        form_data = session.get('form_data', {})
+
+        service_name = form_data.get('service_name', '')
+        sections = session.get('sections', [])
+        current_section = 'description'  
+
+        if sections:
+            current_section = sections[-1]['name']  
+
+        section_order = ['description', 'business_model', 'setup_process', 'budget']
+        current_section_index = section_order.index(current_section) if current_section in section_order else 0
+        progress = int(((current_section_index + 1) / len(section_order)) * 100)
+        next_section_index = section_order.index(current_section) + 1
+
+
+        if 'accept' in request.form and sections:
+            # Find the next section based on the current_section
+            section_order = ['description', 'business_model', 'setup_process', 'budget']
+            next_section_index  = section_order.index(current_section) + 1
+            if next_section_index  < len(section_order):
+                current_section = section_order[next_section_index ]
+                prompt_template = create_prompt_template(current_section, session['form_data'])
+                response = generate_description(prompt_template, get_generation_config(), get_safety_settings())
+                formatted_response = format_response(response)
+                sections.append({'name': current_section, 'content': formatted_response})
+            else:
+                # All sections are complete
+                return render_template('complete.html', sections=sections, service_name=service_name)
+        elif 'regenerate' in request.form:
+            # Regenerate current section
+            prompt_template = create_prompt_template(current_section, session['form_data'])
+            response = generate_description(prompt_template, get_generation_config(), get_safety_settings())
+            formatted_response = format_response(response)
+            if sections:
+                sections[-1]['content'] = formatted_response
+            else:
+                sections.append({'name': current_section, 'content': formatted_response})
+        else:
+            # First visit or generating first section
+            prompt_template = create_prompt_template(current_section, session['form_data'])
+            response = generate_description(prompt_template, get_generation_config(), get_safety_settings())
+            formatted_response = format_response(response)
+            sections.append({'name': current_section, 'content': formatted_response})
+
+        
+        
+        prompt_template = create_prompt_template(current_section, session['form_data'])
+        response = generate_description(prompt_template, get_generation_config(), get_safety_settings())
+        formatted_response = format_response(response)
+
+        # Update or add the current section's content
+        if sections and 'accept' not in request.form:
+            sections[-1]['content'] = formatted_response  # Update last section if regenerating
+        else:
+            sections.append({'name': current_section, 'content': formatted_response})  # Add new section
+
+        # Update session data
+        session['sections'] = sections
+        progress = calculate_progress(sections, section_order)
+
+        return render_template('generate.html', sections=sections, current_section=current_section, service_name=service_name, progress=progress)
+
+
+
+
+    else:
+        # Clear the session for a new start
+        session.clear()
+        return render_template('generate.html', sections=[], current_section='description', service_name='', progress=0)
+
+
+
+def create_prompt_template(current_section, form_data):
+    # Create a detailed prompt for the current section based on form_data
+    # Adjust this function as necessary to tailor prompts for each section
+    prompt_parts = [
+        f"Generate {current_section} for {form_data['service_name']}",
+        f"Land Size: {form_data['land_size']} hectares",
+        f"Biodiversity: {form_data['biodiversity']}",
+        f"Budget: INR {form_data['budget']}",
+        f"Existing Infrastructure: {', '.join(form_data['infrastructure'])}.",
+    ]
+    return " ".join(prompt_parts)
+
+
+if __name__ == '__main__':
+    app.run(debug=True)
+
diff --git a/templates/generate.html b/templates/generate.html
index df971f1..83ffbda 100644
--- a/templates/generate.html
+++ b/templates/generate.html
@@ -32,88 +32,79 @@
       <h2 class="text-xl font-bold text-green-600">
         Agrotourism Service Description Generator
       </h2>
+
       <div class="my-4">
         <div class="flex items-center mb-2">
-          <div class="text-sm font-medium text-green-500 leading-none mr-2">
-            Progress:
-          </div>
-          <div class="flex-1 h-2 bg-gray-200 rounded-full">
-            <div
-              class="h-2 bg-green-500 rounded-full"
-              style="width: {{ '%d%%' | format(progress) }}"
-            ></div>
-          </div>
+            <div class="text-sm font-medium text-green-500 leading-none mr-2">
+                Progress:
+            </div>
+            <div class="flex-1 h-2 bg-gray-200 rounded-full">
+                <div class="h-2 bg-green-500 rounded-full" style="width: {{ progress }}%;"></div>
+            </div>
         </div>
         <div class="flex justify-between text-xs font-medium text-gray-600">
-          <div
-            class="{{ 'text-green-600' if progress >= 25 else 'text-gray-600' }}"
-          >
-            Description
-          </div>
-          <div
-            class="{{ 'text-green-600' if progress >= 50 else 'text-gray-600' }}"
-          >
-            Business Model
-          </div>
-          <div
-            class="{{ 'text-green-600' if progress >= 75 else 'text-gray-600' }}"
-          >
-            Setup Process
-          </div>
-          <div
-            class="{{ 'text-green-600' if progress == 100 else 'text-gray-600' }}"
-          >
-            Budget
-          </div>
+            <div class="{{ 'text-green-600' if progress >= 25 else 'text-gray-600' }}">Description</div>
+            <div class="{{ 'text-green-600' if progress >= 50 else 'text-gray-600' }}">Business Model</div>
+            <div class="{{ 'text-green-600' if progress >= 75 else 'text-gray-600' }}">Setup Process</div>
+            <div class="{{ 'text-green-600' if progress == 100 else 'text-gray-600' }}">Budget</div>
         </div>
-      </div>
+    </div>
       <form method="post" class="mt-4">
-        <label
-          for="service_name"
-          class="block text-gray-700 text-sm font-bold mb-2"
-          >Enter the Agrotourism Service Name:</label
-        >
-        <input
-          type="text"
-          id="service_name"
-          name="service_name"
-          required
-          value="{{ service_name }}"
-          class="shadow appearance-none border rounded py-2 px-3 text-gray-700 leading-tight focus:outline-none focus:shadow-outline w-full"
-        />
-        <input
-          type="submit"
-          value="Start Generation"
-          class="mt-4 bg-green-500 hover:bg-green-700 text-white font-bold py-2 px-4 rounded focus:outline-none focus:shadow-outline cursor-pointer"
-        />
-      </form>
+        <input type="hidden" name="current_section" value="{{ current_section }}">
+        <input type="hidden" name="service_name" value="{{ service_name }}">
+            <div class="form-section">
+                <label for="service_name" class="block text-gray-700 text-sm font-bold mb-2">Enter the Agrotourism Service Name:</label>
+                <input type="text" id="service_name" name="service_name" required class="form-input" />
+            </div>
+
+            <div class="form-section">
+                <label class="font-semibold text-lg mb-2 block">Land Size</label>
+                <input type="number" name="land_size" class="form-input" placeholder="Enter land size in hectares" aria-label="Land Size" required>
+            </div>
+
+            <div class="form-section">
+                <label class="font-semibold text-lg mb-2 block">Biodiversity</label>
+                <select name="biodiversity" class="form-select" aria-label="Biodiversity">
+                    <option value="coastal">Coastal</option>
+                    <option value="forest">Forest</option>
+                    <option value="hills">Hills</option>
+                    <option value="plains">Plains</option>
+                </select>
+            </div>
+
+            <div class="form-section">
+                <label class="font-semibold text-lg mb-2 block">Budget (INR)</label>
+                <input type="number" name="budget" class="form-input" placeholder="Enter your budget in INR" aria-label="Budget" required>
+            </div>
+
+            <div class="form-section">
+                <label class="font-semibold text-lg mb-2 block">Existing Infrastructure</label>
+                <div class="checkbox-container">
+                    <label><input type="checkbox" name="infrastructure[]" value="Tea Bungalow" class="form-checkbox mr-2">Tea Bungalow</label>
+                    <label><input type="checkbox" name="infrastructure[]" value="Tent Accommodation" class="form-checkbox mr-2">Tent Accommodation</label>
+                    <label><input type="checkbox" name="infrastructure[]" value="Cottage" class="form-checkbox mr-2">Cottage</label>
+                    <label><input type="checkbox" name="infrastructure[]" value="Pond" class="form-checkbox mr-2">Pond</label>
+                    <label><input type="checkbox" name="infrastructure[]" value="Storage Facility" class="form-checkbox mr-2">Storage Facility</label>
+                    <label><input type="checkbox" name="infrastructure[]" value="Traditional House" class="form-checkbox mr-2">Traditional House</label>
+                </div>
+            </div>
+
+            <input type="submit" value="Start Generation" class="bg-green-500 hover:bg-green-700 text-white font-bold py-2 px-4 rounded focus:outline-none focus:shadow-outline cursor-pointer" />
+          </form>
       {% if sections %}
       <div class="mt-8">
         {% for section in sections %}
         <div class="mb-8 p-4 shadow-lg rounded-lg bg-white">
-          <h3 class="font-bold text-green-600">
-            Generated {{ section.name.replace('_', ' ').title() }} for {{
-            service_name }}:
-          </h3>
+          <h3 class="font-bold text-green-600">Generated {{ section.name.replace('_', ' ').title() }} for {{ service_name }}:</h3>
           <div class="mt-2 text-gray-600">{{ section.content | safe }}</div>
-        </div>
+      </div>
         {% endfor %} {% if not is_final_section %}
         <form method="post" class="flex justify-between items-center mt-4">
-          <input type="hidden" name="service_name" value="{{ service_name }}" />
-          <input type="hidden" name="section" value="{{ sections[-1].name }}" />
-          <input
-            type="submit"
-            name="accept"
-            value="Yes, loved it!"
-            class="bg-green-500 hover:bg-green-700 text-white font-bold py-2 px-4 rounded focus:outline-none focus:shadow-outline cursor-pointer"
-          />
-          <input
-            type="submit"
-            name="regenerate"
-            value="Generate Again"
-            class="ml-2 bg-gray-300 hover:bg-gray-400 text-gray-800 font-bold py-2 px-4 rounded focus:outline-none focus:shadow-outline cursor-pointer"
-          />
-        </form>
+          <input type="hidden" name="current_section" value="{{ current_section }}">
+          <input type="hidden" name="service_name" value="{{ service_name }}">
+          <input type="submit" name="accept" value="Yes, loved it!" class="bg-green-500 hover:bg-green-700 text-white font-bold py-2 px-4 rounded focus:outline-none focus:shadow-outline cursor-pointer" />
+          <input type="submit" name="regenerate" value="Generate Again" class="ml-2 bg-gray-300 hover:bg-gray-400 text-gray-800 font-bold py-2 px-4 rounded focus:outline-none focus:shadow-outline cursor-pointer" />
+      </form>
         {% else %}
         <p class="mt-4 text-green-600">Thank you for using our service!</p>
         {% endif %}
@@ -122,21 +113,19 @@ <h3 class="font-bold text-green-600">
       
       {% if is_final_section %}
       <!-- Display Button -->
-      <div class="flex justify-end mt-4">
-        <button
-          id="displayButton"
-          class="bg-blue-500 hover:bg-blue-700 text-white font-bold py-2 px-4 rounded focus:outline-none focus:shadow-outline cursor-pointer"
-        >
-          Display
-        </button>
-      </div>
+      <div class="flex justify-end mt-4 space-x-4">
+        <button id="displayButton" class="bg-blue-500 hover:bg-blue-700 text-white font-bold py-2 px-4 rounded focus:outline-none focus:shadow-outline cursor-pointer">Display Plan</button>
+        <form method="POST" action="/download" class="inline">
+            <input type="hidden" name="service_name" value="{{ service_name }}">
+            <button type="submit" class="bg-green-500 hover:bg-green-700 text-white font-bold py-2 px-4 rounded focus:outline-none focus:shadow-outline cursor-pointer">Download Plan</button>
+        </form>
+    </div>
 
       <!-- Your Agrotourism Service Plan Section, hidden by default -->
-      <div id="approvedSections" class="mt-8 hidden">
+      <!-- <div id="approvedSections" class="mt-8 hidden">
         <h3 class="text-xl font-bold text-green-600">
           Your Agrotourism Service Plan:
         </h3>
-        <!-- Approved Sections will be displayed here -->
         {% for section_name, section_content in approved_sections.items() %}
         <div class="mb-8 p-4 shadow-lg rounded-lg bg-white">
           <h4 class="font-bold text-green-600">
@@ -145,27 +134,23 @@ <h4 class="font-bold text-green-600">
           <div class="mt-2 text-gray-600">{{ section_content | safe }}</div>
         </div>
         {% endfor %}
-      </div>
+      </div> -->
 
-      <form method="POST" action="/download">
-        <input type="hidden" name="service_name" value="{{ service_name }}">
-        <button type="submit" class="bg-green-500 hover:bg-green-700 text-white font-bold py-2 px-4 rounded focus:outline-none focus:shadow-outline cursor-pointer">
-            Download
-        </button>
-    </form>
+      
 
       {% endif %}
     </div>
     <script>
       function toggleMenu() {
         var navigation = document.querySelector(".nav-links");
-        navigation.classList.toggle("mobile-menu");
-      }
-      
-        document.getElementById('displayButton').addEventListener('click', function() {
+        navigation.classList.toggle("hidden"); // Adjusted to toggle hidden class
+    }
+
+    document.getElementById('displayButton')?.addEventListener('click', function() {
         var approvedSections = document.getElementById('approvedSections');
         approvedSections.classList.toggle('hidden');
     });
+
     </script>
   </body>
 </html>
diff --git a/templates/index.html b/templates/index.html
index f7c1ed9..b5d526d 100644
--- a/templates/index.html
+++ b/templates/index.html
@@ -42,7 +42,7 @@
     <div class="hero-section text-center py-20 bg-green-50">
       <h1 class="text-5xl font-bold">Generate the optimal agrotourism plan for your farm with AI</h1>
       <p class="text-xl mt-4">Empower your farm's potential with cutting-edge AI technology.</p>
-      <img src="https://i.postimg.cc/fLHL7zb8/hero.png" alt="Hero Image" class="mx-auto mt-8">
+      <img src="https://i.postimg.cc/RZdBhp6Q/herocropped.png" alt="Hero Image" class="mx-auto mt-8">
   </div>
 
 <section class="py-10">