@@ -24,13 +24,21 @@ type Chat = {
2424 answer ?: string ;
2525 loading ?: string ;
2626} ;
27+
28+ const defaultServiceType : string = localStorage . getItem ( "service" ) || "text" ;
29+ const defaultBackendType : String = localStorage . getItem ( "backend" ) || "java" ;
30+
2731const Content = ( ) => {
2832 const [ update , setUpdate ] = useState < Array < object > > ( [ ] ) ;
2933 const [ busy , setBusy ] = useState < boolean > ( false ) ;
30- const [ summaryResults , setSummaryResults ] = useState < string | null > ( "" ) ;
34+ const [ summaryResults , setSummaryResults ] = useState < string > ( "" ) ;
3135 const [ summaryPrompt , setSummaryPrompt ] = useState < string > ( ) ;
32- const [ serviceType , setServiceType ] = useState < ServiceTypes > ( "text" ) ;
33- const [ backendType , setBackendType ] = useState < BackendTypes > ( "java" ) ;
36+ const [ serviceType , setServiceType ] = useState < ServiceTypes > (
37+ defaultServiceType as ServiceTypes
38+ ) ;
39+ const [ backendType , setBackendType ] = useState < BackendTypes > (
40+ defaultBackendType as BackendTypes
41+ ) ;
3442 const [ settingsOpened , setSettingsOpened ] = useState < boolean > ( false ) ;
3543 const question = useRef < string > ( ) ;
3644 const chatData = useRef < Array < object > > ( [ ] ) ;
@@ -55,7 +63,6 @@ const Content = () => {
5563 if ( Q ) {
5664 if ( x > 0 ) tempArray . pop ( ) ;
5765 tempArray . push ( { question : JSON . parse ( Questions ) [ x ] } ) ;
58- // tempArray.push({ loading: "loading" });
5966 Q = false ;
6067 x ++ ;
6168 } else {
@@ -82,17 +89,32 @@ const Content = () => {
8289 chatData
8390 ) ;
8491 } else {
85- setClient ( InitStomp ( setBusy , setUpdate , messagesDP , chatData ) ) ;
92+ setClient (
93+ InitStomp ( setBusy , setUpdate , messagesDP , chatData , serviceType )
94+ ) ;
8695 }
87- console . log ( "Running Gen AI " ) ;
96+ console . log ( "Running Generative service " ) ;
8897 return ;
8998 case "sim" :
9099 runSimulation ( ) ;
91- console . log ( "running simulation" ) ;
100+ console . log ( "Running simulation" ) ;
92101 return ;
93102 case "summary" :
94- // initWebSocket();
95- console . log ( "summary loading" ) ;
103+ if ( backendType === "python" ) {
104+ initWebSocket (
105+ setSummaryResults ,
106+ setBusy ,
107+ setUpdate ,
108+ messagesDP ,
109+ socket ,
110+ chatData
111+ ) ;
112+ } else {
113+ setClient (
114+ InitStomp ( setBusy , setUpdate , messagesDP , chatData , serviceType )
115+ ) ;
116+ }
117+ console . log ( "Running Summarization service" ) ;
96118 return ;
97119 }
98120 return ( ) => {
@@ -114,7 +136,6 @@ const Content = () => {
114136 autoTimeout : "on" ,
115137 } ,
116138 ] ;
117- //alert("Still waiting for an answer! Hang in there a little longer.");
118139 return ;
119140 }
120141 if ( event . detail . value ) {
@@ -165,31 +186,17 @@ const Content = () => {
165186 } ;
166187
167188 const serviceTypeChangeHandler = ( service : ServiceTypes ) => {
189+ localStorage . setItem ( "service" , service ) ;
168190 setUpdate ( [ ] ) ;
169191 chatData . current = [ ] ;
170192 setServiceType ( service ) ;
171- //toggleDrawer();
172193 } ;
173194 const backendTypeChangeHandler = ( backend : BackendTypes ) => {
174195 setUpdate ( [ ] ) ;
175196 chatData . current = [ ] ;
176197 setBackendType ( backend ) ;
177- switch ( backend ) {
178- case "java" :
179- setClient ( InitStomp ( setBusy , setUpdate , messagesDP , chatData ) ) ;
180- return ;
181- case "python" :
182- initWebSocket (
183- setSummaryPrompt ,
184- setBusy ,
185- setUpdate ,
186- messagesDP ,
187- socket ,
188- chatData
189- ) ;
190- return ;
191- }
192- //toggleDrawer();
198+ localStorage . setItem ( "backend" , backend ) ;
199+ location . reload ( ) ;
193200 } ;
194201
195202 const clearSummary = ( ) => {
@@ -199,6 +206,9 @@ const Content = () => {
199206 const updateSummaryPrompt = ( val : string ) => {
200207 setSummaryPrompt ( val ) ;
201208 } ;
209+ const updateSummaryResults = ( summary : string ) => {
210+ setSummaryResults ( summary ) ;
211+ } ;
202212
203213 return (
204214 < div class = "oj-web-applayout-max-width oj-web-applayout-content oj-flex oj-sm-flex-direction-column demo-bg-main" >
@@ -221,23 +231,18 @@ const Content = () => {
221231 position = "top"
222232 onojClose = { handleToastClose }
223233 > </ oj-c-message-toast >
224- { /* <h1 class="oj-typography-heading-lg oj-flex-bar-start"> </h1> */ }
225234 < div class = "oj-flex-bar-end oj-color-invert demo-header-end" >
226- { /* <h6 class="oj-sm-margin-2x-end">{connState}</h6> */ }
227235 < oj-button onojAction = { toggleDrawer } label = "Toggle" display = "icons" >
228236 < span slot = "startIcon" class = "oj-ux-ico-menu" > </ span >
229237 </ oj-button >
230238 </ div >
231239 </ div >
232240 { serviceType === "text" && (
233- < >
234- { /* <oj-button onojAction={sendPrompt}>Send Prompt</oj-button> */ }
235- < Chat
236- data = { update }
237- question = { question }
238- questionChanged = { handleQuestionChange }
239- />
240- </ >
241+ < Chat
242+ data = { update }
243+ question = { question }
244+ questionChanged = { handleQuestionChange }
245+ />
241246 ) }
242247 { serviceType === "sim" && (
243248 < Simulation
@@ -249,9 +254,11 @@ const Content = () => {
249254 { serviceType === "summary" && (
250255 < Summary
251256 fileChanged = { handleFileUpload }
257+ summaryChanged = { updateSummaryResults }
252258 summary = { summaryResults }
253259 clear = { clearSummary }
254260 prompt = { updateSummaryPrompt }
261+ backendType = { backendType }
255262 />
256263 ) }
257264 </ div >
0 commit comments