Update app.py

app.py

@@ -128,6 +128,7 @@ def StreamLLMChatResponse(prompt):
                     res_box.markdown(f'*{result}*')
             except:
                 st.write(' ')
+        return collected_messages
     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).')
 
@@ -388,7 +389,7 @@ def get_zip_download_link(zip_file):
     return href
 
 
-API_URL = f'https://tonpixzfvq3791u9.us-east-1.aws.endpoints.huggingface.cloud'
+API_URL_IE = f'https://tonpixzfvq3791u9.us-east-1.aws.endpoints.huggingface.cloud'
 headers = {
 	"Authorization": "Bearer XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
 	"Content-Type": "audio/wav"
@@ -397,7 +398,7 @@ headers = {
 def query(filename):
     with open(filename, "rb") as f:
         data = f.read()
-    response = requests.post(API_URL, headers=headers, data=data)
+    response = requests.post(API_URL_IE, headers=headers, data=data)
     return response.json()
 
 def generate_filename(prompt, file_type):
@@ -422,7 +423,6 @@ def transcribe_audio(filename):
     output = query(filename)
     return output
 
-
 def whisper_main():
     st.title("Speech to Text")
     st.write("Record your speech and get the text.")
@@ -432,39 +432,10 @@ def whisper_main():
     if filename is not None:
         transcription = transcribe_audio(filename)
         st.write(transcription)
-        StreamLLMChatResponse(transcription)
-        
-        # Define button descriptions
-        descriptions = {
-            "Generate Limericks: 😂": "Write ten random adult limericks based on quotes that are tweet length and make you laugh 🎭",
-            "Wise Quotes: 🧙": "Generate ten wise quotes that are tweet length 🦉",
-            "Funny Rhymes: 🎤": "Create ten funny rhymes that are tweet length 🎶",
-            "Medical Jokes: 💉": "Create ten medical jokes that are tweet length 🏥",
-            "Minnesota Humor: ❄️": "Create ten jokes about Minnesota that are tweet length 🌨️",
-            "Top Funny Stories: 📖": "Create ten funny stories that are tweet length 📚",
-            "More Funny Rhymes: 🎙️": "Create ten more funny rhymes that are tweet length 🎵"
-        }
-        
-        # Create columns
-        col1, col2, col3 = st.columns([1, 1, 1], gap="small")
-        
-        # Add buttons to columns
-        if col1.button("Generate Limericks: 😂"):
-            StreamLLMChatResponse(descriptions["Generate Limericks: 😂"])
-        if col2.button("Wise Quotes 🧙"):
-            StreamLLMChatResponse(descriptions["Wise Quotes: 🧙"])
-        if col3.button("Funny Rhymes: 🎤"):
-            StreamLLMChatResponse(descriptions["Funny Rhymes: 🎤"])
-        col4, col5, col6 = st.columns([1, 1, 1], gap="small")
-        if col4.button("Medical Jokes: 💉"):
-            StreamLLMChatResponse(descriptions["Medical Jokes: 💉"])
-        if col5.button("Minnesota Humor ❄️"):
-            StreamLLMChatResponse(descriptions["Minnesota Humor: ❄️"])
-        if col6.button("Top Funny Stories 📖"):
-            StreamLLMChatResponse(descriptions["Top Funny Stories: 📖"])
-        col7 = st.columns(1, gap="small")
-        if col7[0].button("More Funny Rhymes: 🎙️"):
-            StreamLLMChatResponse(descriptions["More Funny Rhymes: 🎙️"])
+        response = StreamLLMChatResponse(transcription)                            
+        filename = generate_filename(transcription, choice)
+        create_file(filename, user_prompt, response, should_save)
+        st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
 
 def main():