Update app.py

app.py

@@ -17,11 +17,13 @@ model = ChatOpenAI(model_name="gpt-4-0125-preview", temperature=0.4)
 
 # System prompt templates
 raft_base_template  = os.getenv('RAFT_BASE_TEMPLATE')
+raft_base_template_v2  = os.getenv('RAFT_BASE_TEMPLATE_V2')
+raft_metaprompt_template = os.getenv('RAFT_METAPROMPT_TEMPLATE')
+raft_metaprompt_template_v2 = os.getenv('RAFT_METAPROMPT_TEMPLATE_2')
 
 # System prompt template
 customer_template = os.getenv('CUSTOMER_TEMPLATE')
 
-raft_metaprompt_template = os.getenv('RAFT_METAPROMPT_TEMPLATE')
 # Initialize DataFrame
 results_df = pd.DataFrame(columns=['Experiment Number','Temperature', 'PromptID','Business Idea', 'Generated Domains'])
 
@@ -37,11 +39,11 @@ def extract_content_inside_brackets(json_string):
     else:
         # Возвращаем None, если скобки не найдены
         return None
-list_of_prompts = ['customer_template', 'raft_base_template', 'raft_metaprompt_template']
+list_of_prompts = ['customer_template', 'raft_base_template', 'raft_metaprompt_template', 'raft_base_template_v2', 'raft_metaprompt_template_v2']
 # Function to generate domain names based on different templates
 async def generate_domain_names(business_idea, temperature):
     model.temperature = temperature  # Update model temperature
-    prompts = [customer_template, raft_base_template, raft_metaprompt_template]
+    prompts = [customer_template, raft_base_template, raft_metaprompt_template, raft_base_template_v2, raft_metaprompt_template_v2]
     tasks = []
     for prompt in prompts:
         system_message_prompt = SystemMessagePromptTemplate.from_template(prompt)
@@ -89,12 +91,14 @@ def setup_interface():
             output1 = gr.Textbox(label="Output for Customer Template")
             output2 = gr.Textbox(label="Output for Base Raft Template")
             output3 = gr.Textbox(label="Output for MetaPrompt Raft Template")
+            output4 = gr.Textbox(label="Output for Base Raft Template (новый домены)")
+            output5 = gr.Textbox(label="Output for MetaPrompt Raft Template (новый домены)")
         with gr.Row():
             download_btn = gr.Button("Download Excel")
         submit_button.click(
             fn=generate_domain_names,
             inputs=[text_input, temperature_slider],
-            outputs=[output1, output2, output3]
+            outputs=[output1, output2, output3, output4, output5]
         )
         download_btn.click(
             fn=download_excel,