Update chat_interface_preference.py

chat_interface_preference.py

@@ -21,6 +21,7 @@ from gradio.components import (
     Component,
     Markdown,
     MultimodalTextbox,
+    Slider,
     State,
     Textbox,
     get_component_instance,
@@ -184,12 +185,16 @@ class ChatInterface(Blocks):
         if self.commit_scheduler:
             self.data_file = self.commit_scheduler.folder_path / f"data_{uuid.uuid4()}.json"
 
+        slider_placeholder = Slider(label="generation", minimum=1.0, maximum=2.0, step=0.05, value=1.2, render=False)
+
         if additional_inputs:
             if not isinstance(additional_inputs, list):
-                additional_inputs = [additional_inputs]
+                additional_inputs = [slider_placeholder] + [additional_inputs]
+            else:
+                additional_inputs = [slider_placeholder] + additional_inputs
             self.additional_inputs = [get_component_instance(i) for i in additional_inputs]  # type: ignore
         else:
-            self.additional_inputs = []
+            self.additional_inputs = [slider_placeholder]
         if additional_inputs_accordion_name is not None:
             print(
                 "The `additional_inputs_accordion_name` parameter is deprecated and will be removed in a future version of Gradio. Use the `additional_inputs_accordion` parameter instead."
@@ -382,10 +387,11 @@ class ChatInterface(Blocks):
 
     def _setup_events(self) -> None:
         submit_fn_one = self._stream_fn if self.is_generator else self._submit_fn
+        submit_fn_one_partial = functools.partial(submit_fn_one, n_generations=1)
         submit_triggers_one = (
             [self.textbox.submit, self.submit_btn_one.click] if self.submit_btn_one else [self.textbox.submit]
         )
-        submit_tuples = [(submit_fn_one, submit_triggers_one)]
+        submit_tuples = [(submit_fn_one_partial, submit_triggers_one)]
         if self.submit_btn_two:
             submit_fn_two = functools.partial(submit_fn_one, n_generations=2)
             submit_triggers_two = [self.submit_btn_two.click]
@@ -671,8 +677,8 @@ class ChatInterface(Blocks):
         message: str | dict[str, list],
         history_with_input: list[list[str | tuple | None]],
         request: Request,
-        n_generations: int = 1,
         *args,
+        n_generations: int,
     ) -> tuple[list[list[str | tuple | None]], list[list[str | tuple | None]]]:
         if self.multimodal and isinstance(message, dict):
             remove_input = len(message["files"]) + 1 if message["text"] is not None else len(message["files"])
@@ -712,23 +718,20 @@ class ChatInterface(Blocks):
         message: str | dict[str, list],
         history_with_input: list[list[str | tuple | None]],
         request: Request,
-        n_generations: int = 1,
         *args,
+        n_generations: int,
     ) -> AsyncGenerator:
         if self.multimodal and isinstance(message, dict):
             remove_input = len(message["files"]) + 1 if message["text"] is not None else len(message["files"])
             history = history_with_input[:-remove_input]
         else:
             history = history_with_input[:-1]
-
         self._check_message(message)
         self._check_num_turns(history)
         _, response = history_with_input[-1]
         if self._check_if_two_responses(response):
             raise Error("Two options detected: undo, log or random pick continuation.")
 
-        _, response = history_with_input[-1]
-
         inputs, _, _ = special_args(self.fn, inputs=[message, history, *args], request=request)
 
         try:
@@ -900,10 +903,13 @@ class ChatInterface(Blocks):
         str | dict[str, list],
         list[list[str | tuple | None]],
     ]:
-        _, response = history[-1]
-        if self._check_if_two_responses(response):
-            raise Error("First log preference or continue random.")
+        if history:
+            _, response = history[-1]
+            if self._check_if_two_responses(response):
+                raise Error("First log preference or continue random.")
+            else:
+                await self._log_fn(message=message, history=history, log="prompt")
+                self._set_conversation_id()
+                return [], "", []
         else:
-            await self._log_fn(message=message, history=history, log="prompt")
-            self._set_conversation_id()
             return [], "", []