Update content_generator.py

content_generator.py

@@ -34,16 +34,15 @@ def get_website_content(row, query_engine):
     try:
         if pd.notna(row['Suggested Word/Character Count']):
             if row['Suggested Word/Character Count'] != 'Not applicable for SEO':
-              if row['Prompt']:
-                  prompt = row['Prompt']
-              else:
-                  prompt = None
               page = row['Page Type']
               instructions = row['Suggested Word/Character Count']
               pov = row['Point of View']
               element = row['Section Piece (Element)']
-              
-              response = query_engine.query(get_website_layout_prompt(page, instructions, pov, element, csv_prompt=prompt))
+              if pd.notna(row['Prompt']):
+                  prompt = row['Prompt']
+                  response = query_engine.query(get_website_layout_prompt(page, instructions, pov, element, csv_prompt=prompt))
+              else:
+                  response = query_engine.query(get_website_layout_prompt(page, instructions, pov, element))
               print(response)
               return response.response
             else: