Update app.py

app.py

@@ -115,7 +115,7 @@ def StreamLLMChatResponse(prompt):
         res_box = st.empty()
         collected_chunks=[]
         collected_messages=[]
-        #allresults=''
+        allresults=''
         for r in stream:
             if r.token.special:
                 continue
@@ -129,12 +129,10 @@ def StreamLLMChatResponse(prompt):
                 if len(r.token.text) > 0:
                     result="".join(report).strip()
                     res_box.markdown(f'*{result}*')
-                    #allresults=allresults+result
+                    
             except:
                 st.write('Stream llm issue')
-
-        #st.write(allresults)
-        #return allresults
+        return result
     except:
         st.write('DromeLlama is asleep. Starting up now on A10 - please give 5 minutes then retry as KEDA scales up from zero to activate running container(s).')
 
@@ -195,7 +193,7 @@ def create_file(filename, prompt, response, should_save=True):
     has_python_code = bool(re.search(r"```python([\s\S]*?)```", response))
     if ext in ['.txt', '.htm', '.md']:
         with open(f"{base_filename}-Prompt.txt", 'w') as file:
-            file.write(prompt)
+            file.write(prompt.strip())
         with open(f"{base_filename}-Response.md", 'w') as file:
             file.write(response)
         if has_python_code:
@@ -437,12 +435,12 @@ def whisper_main():
     filename = save_and_play_audio(audio_recorder)
     if filename is not None:
         transcription = transcribe_audio(filename)
+        transcription = transcription['text']
         st.write(transcription)
-        response = StreamLLMChatResponse(transcription['text'])
-        st.write(response)
-        #filename = generate_filename(str(response), ".txt")
-        #st.write(filename)
-        #create_file(filename, transcription, '', True)
+        response = StreamLLMChatResponse(transcription)
+        # st.write(response) - redundant with streaming result?
+        filename = generate_filename(transcription, ".txt")
+        create_file(filename, transcription, response, should_save)
         #st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
 
 def main():