Add code separation getting generators

chat_interface_preference.py

@@ -641,16 +641,12 @@ class ChatInterface(Blocks):
 
         inputs, _, _ = special_args(self.fn, inputs=[message, history, *args], request=request)
 
-        async def _get_response():
+        try:
             if self.is_async:
                 generator = self.fn(*inputs)
             else:
                 generator = await anyio.to_thread.run_sync(self.fn, *inputs, limiter=self.limiter)
                 generator = SyncToAsyncIterator(generator, self.limiter)
-            return generator
-
-        generator = await _get_response()
-        try:
             first_response = await async_iteration(generator)
             if n_generations == 2:
                 first_response_formatted = self._get_chat_message_comparison(first_response, "")
@@ -681,7 +677,11 @@ class ChatInterface(Blocks):
                 yield update, update
 
         if n_generations == 2:
-            generator_two = await _get_response()
+            if self.is_async:
+                generator_two = self.fn(*inputs)
+            else:
+                generator_two = await anyio.to_thread.run_sync(self.fn, *inputs, limiter=self.limiter)
+                generator_two = SyncToAsyncIterator(generator, self.limiter)
             try:
                 first_response_two = await async_iteration(generator_two)
                 first_response_two = self._get_chat_message_comparison(response, first_response_two)