Update app.py

app.py

@@ -22,7 +22,7 @@ class GUI:
             inp.change(self.process, inp, out)
             self.ws = None
             self.out = []
-        demo.launch()
+        demo.queue(concurrency_count=2).launch()
 
     def process(self, file, profile: gr.OAuthProfile | None):
         if profile is None:
@@ -32,20 +32,22 @@ class GUI:
         chapter_titles = epub2txt.content_titles
         title = epub2txt.title
         if self.ws is None:
-            self.ws = create_connection(f"ws://localhost/{file.name}/ws")
-        res = self.ws.recv()
-        if 'chsum: ' in res:
-            self.out.append(res.remove("chsum: "))
-        elif 'draft_sum: ' in res:
-            h = res[11:]
-        elif 'output: ' in res:
-            self.ws.close()
-            self.ws = None
-            self.out = []
-            return gr.update(value=res)
-        return gr.update(
-            value=f"# {title}\n\n" +h+ "\n\n".join(
-                [f"## {ct}\n\n{c}" for ct, c in zip(chapter_titles, self.out)]))
+            self.ws = create_connection(f"ws://localhost:8000/{file.name}/ws")
+        res = ''
+        while 'output: ' not in res:
+            res = self.ws.recv()
+            if 'chsum: ' in res:
+                self.out.append(res.remove("chsum: "))
+            elif 'draft_sum: ' in res:
+                h = res[11:]
+            elif 'output: ' in res:
+                self.ws.close()
+                self.ws = None
+                self.out = []
+                yield gr.update(value=res)
+            yield gr.update(
+                value=f"# {title}\n\n" +h+ "\n\n".join(
+                    [f"## {ct}\n\n{c}" for ct, c in zip(chapter_titles, self.out)]))
 
     def hello(self, profile: gr.OAuthProfile | None):
         if profile is None: