feat: change interface

app.py

@@ -88,37 +88,246 @@ def generate(
         yield cleaned_response
 
 
-def clear():
-    return "", ""
+with gr.Blocks(
+    css="""
+    .gradio-container { max-width: 1600px; margin: 20px; padding: 10px; }
+    #style-dropdown { flex: 3; }
+    #generate-btn, #clear-btn { flex: 1; max-width: 100px; }
+    .same-height { height: 60px; }
+"""
+) as demo:
+    gr.Markdown("# Bielik Tools - narzędzia dla modelu Bielik v2.3")
 
+    with gr.Column(elem_id="main-content"):
+        with gr.Row():
+            simple_question_btn = gr.Button("Proste Pytanie", variant="primary")
+            formalizer_btn = gr.Button("Formalizator", variant="secondary")
+            judge_btn = gr.Button("Sędzia", interactive=False)
 
-with gr.Blocks() as demo:
-    gr.Markdown("# Bielik Tools - narzędzia dla modelu Bielik v2.3")
-    gr.Markdown("Bielik czeka na Twoje pytanie - zadaj je śmiało i otrzymaj odpowiedź!")
+        # Function to switch tool visibility and update button styles based on the active tool
+        def switch_tool(tool):
+            print(f"Switched to {tool}")
+            return [
+                gr.Button(variant="primary" if tool == "Formalizer" else "secondary"),
+                gr.Button(variant="primary" if tool == "Judge" else "secondary"),
+                gr.Button(
+                    variant="primary" if tool == "Simple Question" else "secondary"
+                ),
+                gr.update(visible=(tool == "Formalizer")),
+                gr.update(visible=(tool == "Judge")),
+                gr.update(visible=(tool == "Simple Question")),
+            ]
+
+        # Simple Question content column
+        with gr.Column(visible=True) as simple_question_column:
+            input_text_sq = gr.Textbox(
+                label="Twoje pytanie",
+                placeholder="Zadaj swoje pytanie tutaj...",
+                lines=5,
+            )
+            with gr.Row():
+                generate_btn_sq = gr.Button("Generuj odpowiedź", interactive=False)
+                clear_btn_sq = gr.Button("Wyczyść", interactive=False)
+            output_text_sq = gr.Textbox(label="Odpowiedź", interactive=False)
+
+            with gr.Accordion("⚙️ Parametry", open=False):
+                temperature_sq = gr.Slider(0, 1, 0.3, step=0.1, label="Temperatura")
+                max_tokens_sq = gr.Slider(
+                    128, 4096, 1024, label="Maksymalna długość odpowiedzi"
+                )
+                top_k_sq = gr.Slider(1, 80, 40, step=1, label="Top K")
+                repetition_penalty_sq = gr.Slider(
+                    0, 2, 1.1, step=0.1, label="Penalizacja powtórzeń"
+                )
+                top_p_sq = gr.Slider(0, 1, 0.95, step=0.05, label="Top P")
+
+            # Update button states based on input and output text changes for interactivity
+            def update_button_states_sq(input_text, output_text):
+                return [
+                    gr.update(interactive=bool(input_text)),
+                    gr.update(interactive=bool(input_text or output_text)),
+                ]
+
+            input_text_sq.change(
+                update_button_states_sq,
+                inputs=[input_text_sq, output_text_sq],
+                outputs=[generate_btn_sq, clear_btn_sq],
+            )
+
+            output_text_sq.change(
+                update_button_states_sq,
+                inputs=[input_text_sq, output_text_sq],
+                outputs=[generate_btn_sq, clear_btn_sq],
+            )
+
+            # Event handlers for button actions to process and clear text
+            generate_btn_sq.click(
+                fn=generate,
+                inputs=[
+                    input_text_sq,
+                    temperature_sq,
+                    max_tokens_sq,
+                    top_k_sq,
+                    repetition_penalty_sq,
+                    top_p_sq,
+                ],
+                outputs=output_text_sq,
+            )
+
+            clear_btn_sq.click(
+                fn=lambda: ("", ""),
+                inputs=None,
+                outputs=[input_text_sq, output_text_sq],
+            )
+
+        with gr.Column(visible=False) as formalizer_column:
+            input_text = gr.Textbox(
+                placeholder="Wpisz tekst tutaj...", label="Twój tekst", lines=5
+            )
+            with gr.Row():
+                gr.Text(
+                    "Wybierz styl tekstu:",
+                    elem_id="style-label",
+                    show_label=False,
+                    elem_classes="same-height",
+                )
+                style_dropdown = gr.Dropdown(
+                    choices=["Formalny", "Nieformalny", "Neutralny"],
+                    elem_id="style-dropdown",
+                    show_label=False,
+                    elem_classes="same-height",
+                )
+                generate_btn = gr.Button(
+                    "Generuj",
+                    interactive=False,
+                    elem_id="generate-btn",
+                    elem_classes="same-height",
+                )
+                clear_btn = gr.Button(
+                    "Wyczyść",
+                    interactive=False,
+                    elem_id="clear-btn",
+                    elem_classes="same-height",
+                )
+            output_text = gr.Textbox(label="Wynik", interactive=False)
+
+            # Mock processing function to simulate text formalization with a delay
+            def mock_processing_function(input_text):
+                import time
+
+                time.sleep(1)  # Simulate processing delay
+                return input_text + " [Formalized]"
 
-    with gr.Row():
-        prompt = gr.Textbox(
-            label="Twoje pytanie", placeholder="Zadaj swoje pytanie tutaj...", lines=10
+            # Update button states based on input and output text changes for interactivity
+            def update_button_states(input_text, output_text):
+                return [
+                    gr.update(interactive=bool(input_text)),
+                    gr.update(interactive=bool(input_text or output_text)),
+                    gr.update(interactive=bool(output_text)),
+                ]
+
+            input_text.change(
+                update_button_states,
+                inputs=[input_text, output_text],
+                outputs=[generate_btn, clear_btn],
+            )
+
+            output_text.change(
+                update_button_states,
+                inputs=[input_text, output_text],
+                outputs=[generate_btn, clear_btn],
+            )
+
+            # Event handlers for button actions to process and clear text
+            generate_btn.click(
+                fn=mock_processing_function, inputs=input_text, outputs=output_text
+            )
+
+            clear_btn.click(
+                fn=lambda: ("", ""), inputs=None, outputs=[input_text, output_text]
+            )
+
+        # Placeholder for Judge content column, initially hidden
+        with gr.Column(visible=False) as judge_column:
+            gr.Markdown("Judge tool content goes here.")
+
+            with gr.Accordion("⚙️ Parametry", open=False):
+                temperature_jg = gr.Slider(0, 1, 0.3, step=0.1, label="Temperatura")
+                max_tokens_jg = gr.Slider(
+                    128, 4096, 1024, label="Maksymalna długość odpowiedzi"
+                )
+                top_k_jg = gr.Slider(1, 80, 40, step=1, label="Top K")
+                repetition_penalty_jg = gr.Slider(
+                    0, 2, 1.1, step=0.1, label="Penalizacja powtórzeń"
+                )
+                top_p_jg = gr.Slider(0, 1, 0.95, step=0.05, label="Top P")
+
+        formalizer_btn.click(
+            lambda: switch_tool("Formalizer"),
+            outputs=[
+                formalizer_btn,
+                judge_btn,
+                simple_question_btn,
+                formalizer_column,
+                judge_column,
+                simple_question_column,
+            ],
+        )
+        judge_btn.click(
+            lambda: switch_tool("Judge"),
+            outputs=[
+                formalizer_btn,
+                judge_btn,
+                simple_question_btn,
+                formalizer_column,
+                judge_column,
+                simple_question_column,
+            ],
         )
-        output = gr.Textbox(label="Answer", lines=10)
-
-    with gr.Row():
-        btn = gr.Button("Generuj odpowiedź")
-        clear_btn = gr.Button("Wyczyść")
-
-    with gr.Accordion("⚙️ Parametry", open=False):
-        temperature = gr.Slider(0, 1, 0.3, step=0.1, label="Temperatura")
-        max_tokens = gr.Slider(128, 4096, 1024, label="Maksymalna długość odpowiedzi")
-        top_k = gr.Slider(1, 80, 40, step=1, label="Top K")
-        repetition_penalty = gr.Slider(
-            0, 2, 1.1, step=0.1, label="Penalizacja powtórzeń"
+        simple_question_btn.click(
+            lambda: switch_tool("Simple Question"),
+            outputs=[
+                formalizer_btn,
+                judge_btn,
+                simple_question_btn,
+                formalizer_column,
+                judge_column,
+                simple_question_column,
+            ],
         )
-        top_p = gr.Slider(0, 1, 0.95, step=0.05, label="Top P")
-    btn.click(
-        generate,
-        inputs=[prompt, temperature, max_tokens, top_k, repetition_penalty, top_p],
-        outputs=output,
+
+    formalizer_btn.click(
+        lambda: switch_tool("Formalizer"),
+        outputs=[
+            formalizer_btn,
+            judge_btn,
+            simple_question_btn,
+            formalizer_column,
+            judge_column,
+            simple_question_column,
+        ],
+    )
+    judge_btn.click(
+        lambda: switch_tool("Judge"),
+        outputs=[
+            formalizer_btn,
+            judge_btn,
+            simple_question_btn,
+            formalizer_column,
+            judge_column,
+            simple_question_column,
+        ],
+    )
+    simple_question_btn.click(
+        lambda: switch_tool("Simple Question"),
+        outputs=[
+            formalizer_btn,
+            judge_btn,
+            simple_question_btn,
+            formalizer_column,
+            judge_column,
+            simple_question_column,
+        ],
     )
-    clear_btn.click(clear, inputs=[], outputs=[prompt, output])
 
-demo.launch()
+demo.queue().launch()