update app

app.py

@@ -41,37 +41,39 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     with gr.Row():
         task = gr.Dropdown(label="Task", choices=list(tasks_datasets.keys()), value=QA)
         language = gr.Dropdown(label="Source Language", choices=languages, value="English")
-        model_type = gr.Dropdown(label="Model Type", choices=["Multilingual", "English-Centric"], value='Multilingual')
+        model_type = gr.Dropdown(label="Model Type", choices=["Multilingual", "Standard"], value='Multilingual')
     config_recommendation = gr.Button("Recommend Configuration")
     with gr.Row():
         config_prompt = gr.Textbox(label="Recommended Configuration", interactive=False,
                             placeholder="Recommended Configuration for this scenerio")
     with gr.Row():
-        with gr.Column(scale=3):
-            gr.Markdown("### Prompt Configuration", elem_id="section-title")  # Adding a small title
+        # examples_selection = gr.Dropdown(["English", "Source"], label="examples", value='English')
+        # output_selection = gr.Dropdown(["English", "Source"], label="output", value='English')
+        with gr.Column(scale=2):
+            gr.Markdown("### Prompt Template", elem_id="section-title")  # Adding a small title
             # Set the same background style across all components
             with gr.Group(elem_id="prompt-background"):
-
                 instruction = gr.Textbox(label="Instruction")
-                zero_shot = gr.Checkbox(label="Zero-shot", value=False)
-                with gr.Accordion("Few Shot - Select Type of Examples ", open=False, visible=True) as few_shot:
-                    dataset = gr.Dropdown(label="Dataset", choices=tasks_datasets[QA], value="XlSum")
-                    num_examples = gr.Slider(label="Number of examples in context", minimum=1, maximum=10, step=1,
-                                             value=3)
-                # Accordion for Prompt Configuration Selection
-                with gr.Accordion(label="Prompt Configuration Selection", open=False):
-                    prefix_selection = gr.Dropdown(["English", "Source"], label="prefix", value='English')
-                    context_selection = gr.Dropdown(["English", "Source"], label="context", value='English')
-                    examples_selection = gr.Dropdown(["English", "Source"], label="examples", value='English')
-                    output_selection = gr.Dropdown(["English", "Source"], label="output", value='English')
-                    # Accordion for Few Shot example selection
-                with gr.Row():
-                    question = gr.Textbox(label="Question", visible=True)
-                    context = gr.Textbox(label="Context", visible=True)
-                    text = gr.Textbox(label="Text", visible=False)
-                    sentence = gr.Textbox(label="Sentence", visible=False)
-                    hypothesis = gr.Textbox(label="Hypothesis", visible=False)
-                    premise = gr.Textbox(label="Premise", visible=False)
+                with gr.Row(variant="panel"):
+                    zero_shot = gr.Checkbox(label="Zero Shot Setting", value=False)
+                    with gr.Accordion("Few Shot - Select Type of Examples ", open=False, visible=True) as few_shot:
+                        dataset = gr.Dropdown(label="Dataset", choices=tasks_datasets[QA], value="XlSum")
+                        num_examples = gr.Slider(label="Number of examples in context", minimum=1, maximum=10, step=1,
+                                            value=3)
+                with gr.Row(equal_height=True, variant="panel"):
+                    with gr.Accordion(label="Language Component Selection", open=False):
+                        prefix_selection = gr.Dropdown(["English", "Source"], label="prefix", value='English')
+                        context_selection = gr.Dropdown(["English", "Source"], label="context", value='English')
+                        examples_selection = gr.Dropdown(["English", "Source"], label="examples", value='English')
+                        output_selection = gr.Dropdown(["English", "Source"], label="output", value='English')
+                        # Accordion for Few Shot example selection
+                    with gr.Accordion(label="Prompt Input Data", open=False):
+                        question = gr.Textbox(label="Question", visible=True)
+                        context = gr.Textbox(label="Context", visible=True)
+                        text = gr.Textbox(label="Text", visible=False)
+                        sentence = gr.Textbox(label="Sentence", visible=False)
+                        hypothesis = gr.Textbox(label="Hypothesis", visible=False)
+                        premise = gr.Textbox(label="Premise", visible=False)
 
     generate_button = gr.Button("Generate Prompt")
 
@@ -214,4 +216,4 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
     )
 
 if __name__ == '__main__':
-    demo.launch(share=True)
+    demo.launch(share=True)

generate_prompt.py

@@ -608,7 +608,7 @@ class Config:
 
     def to_dict(self):
         return {
-            'prefix': self.prefix,
+            'instruction': self.prefix,
             'context': self.context,
             'examples': self.examples,
             'output': self.output
@@ -616,12 +616,8 @@ class Config:
 
 
 def recommend_config(task, lang, model_type):
-    print(task)
-    print(model_type)
-    print(lang)
     language_type = _get_language_type(lang)
     config = Config(lang, lang, lang, lang)
-    print(language_type)
     if task == QA:
         if model_type == ModelType.English.value:
             config.set(prefix=lang, context=lang, examples=lang, output=lang)
@@ -643,5 +639,5 @@ def recommend_config(task, lang, model_type):
             config.set(prefix='English', context='English', examples='English')
     if task == SUMMARIZATION:
         config.set(context='English')
-
+    print(config.to_dict())
     return config.to_dict()