From aa8e700be5d4de37a167db512302c492349b3b9c Mon Sep 17 00:00:00 2001
From: NamH <NamNh0122@gmail.com>
Date: Tue, 5 Dec 2023 18:21:03 +0700
Subject: [PATCH] fix: update wrong main view state when use a model (#861)

Signed-off-by: James <james@jan.ai>
Co-authored-by: James <james@jan.ai>
---
 .../ExploreModels/ExploreModelItemHeader/index.tsx     | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx b/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx
index d152d9b66d..3621c90ec6 100644
--- a/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx
+++ b/web/screens/ExploreModels/ExploreModelItemHeader/index.tsx
@@ -59,15 +59,17 @@ const ExploreModelItemHeader: React.FC<Props> = ({ model, onClick, open }) => {
     <Button onClick={() => onDownloadClick()}>Download</Button>
   )
 
+  const onUseModelClick = () => {
+    startModel(model.id)
+    setMainViewState(MainViewState.Thread)
+  }
+
   if (isDownloaded) {
     downloadButton = (
       <Button
         themes="success"
         className="min-w-[98px]"
-        onClick={() => {
-          startModel(model.id)
-          setMainViewState(MainViewState.MyModels)
-        }}
+        onClick={onUseModelClick}
       >
         Use
       </Button>