fix: add cancel for generate text

2024-12-02 17:50:21 +08:00
parent 804cbb8c56
commit a7456d1182
3 changed files with 77 additions and 50 deletions
@@ -319,13 +319,28 @@ const GroundLeft: React.FC<MessageProps> = forwardRef((props, ref) => {
                      id: 'playground.audio.button.generate'
                    })}
                  >
-                    <Button
-                      disabled={!audioData}
-                      type="primary"
-                      shape="circle"
-                      onClick={handleOnGenerate}
-                      icon={<SendOutlined></SendOutlined>}
-                    ></Button>
+                    {loading ? (
+                      <Button
+                        disabled={!audioData}
+                        type="primary"
+                        shape="circle"
+                        onClick={handleStopConversation}
+                        icon={
+                          <IconFont
+                            type="icon-stop1"
+                            className="font-size-14"
+                          ></IconFont>
+                        }
+                      ></Button>
+                    ) : (
+                      <Button
+                        disabled={!audioData}
+                        type="primary"
+                        shape="circle"
+                        onClick={handleOnGenerate}
+                        icon={<SendOutlined></SendOutlined>}
+                      ></Button>
+                    )}
                  </Tooltip>
                </div>
              </div>