Update app.py

app.py

@@ -139,6 +139,12 @@ def generate_fb_ad(target_audience, product, temperature, selected_formula, sele
         headline=headline  # Pass the headline to the instruction creator
     )
     
+    # Ensure the headline is properly integrated in the instruction
+    if headline:
+        ad_instruction = ad_instruction.replace("[HEADLINE]", headline)
+        # Add explicit instruction to use the headline
+        ad_instruction += f"\n\nUtiliza este titular exactamente como está escrito: '{headline}'"
+        
     # Si hay contenido de archivo, añadirlo a la instrucción
     if file_content:
         ad_instruction += f"\n\nAdemás, utiliza la siguiente información como referencia para crear el anuncio:\n\n{file_content[:max_file_chars]}"
@@ -348,10 +354,17 @@ if submit_ad:
                     st.error("Error al generar el anuncio")
                 else:
                     # Clean the generated ad text to remove any random characters
+                    # First, remove any markdown code block markers
+                    cleaned_ad = generated_ad.replace("```", "").replace("text", "")
+                    
                     # Remove any single letters that appear on their own line
-                    cleaned_ad = "\n".join([line for line in generated_ad.split("\n") 
+                    cleaned_ad = "\n".join([line for line in cleaned_ad.split("\n") 
                                           if not (len(line.strip()) == 1 and line.strip().isalpha())])
                     
+                    # Remove any lines with just symbols or very short content
+                    cleaned_ad = "\n".join([line for line in cleaned_ad.split("\n") 
+                                          if not (len(line.strip()) <= 3 and not line.strip().isalnum())])
+                    
                     # Display only the ad without separate headline and without storing in session state
                     st.markdown(f"""
                         <div style="{styles['results_container']}">