Update app.py

app.py

@@ -208,10 +208,9 @@ with column2:
                         if index==0:
 
                             st.write(prompts_[0].split(send_prompt())[-1].upper() if send_prompt() in prompts_[0] else prompts_[0].upper())
-                            data_need=""
-                            while(len(data_need)==0):
-                                data_need=st.write_stream(consume_llm_api(prompts_[0]))
-                            dictionary['every_prompt_with_val'][-1]=(prompts_[0],str(data_need))
+                            data_need=st.write_stream(consume_llm_api(prompts_[0]))
+                            if len(data_need)!=0:
+                                dictionary['every_prompt_with_val'][-1]=(prompts_[0],str(data_need))
                             
                     elif isinstance(prompts_[-1],str):
                         show_case_text=prompts_[0].split(send_prompt())[-1].upper() if send_prompt() in prompts_[0] else prompts_[0].upper()