Spaces:
Running
Running
File size: 775 Bytes
54ec520 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from .prompt_templates import PromptTemplates
def generate_prompt(focus_type, paragraph, length, sentiment, factuality, language, narrative):
mapped_value = PromptTemplates.FOCUS_MAP.get(focus_type, -1)
narrative_value = PromptTemplates.NARRATIVE_MAPPING[narrative]
controls = {
'length': length,
'sentiment': sentiment,
'factuality': factuality,
'language': language
}
if mapped_value != -1:
prompt = PromptTemplates.ANALYSIS_PROMPTS[narrative_value][mapped_value].format(
Wiki_caption=paragraph,
length=controls['length'],
sentiment=controls['sentiment'],
language=controls['language']
)
else:
prompt = "Invalid focus type."
return prompt |