Update backupapp.py

backupapp.py

@@ -31,7 +31,7 @@ from PyPDF2 import PdfReader
 from templates import bot_template, css, user_template
 from xml.etree import ElementTree as ET
 
-# Constants
+# Llama Constants
 API_URL = 'https://qe55p8afio98s0u3.us-east-1.aws.endpoints.huggingface.cloud'  # Dr Llama
 API_KEY = os.getenv('API_KEY')
 headers = {
@@ -172,7 +172,7 @@ def transcribe_audio(openai_key, file_path, model):
         return None
 
 def save_and_play_audio(audio_recorder):
-    audio_bytes = audio_recorder()
+    audio_bytes = audio_recorder(key='audio_recorder')
     if audio_bytes:
         filename = generate_filename("Recording", "wav")
         with open(filename, 'wb') as f:
@@ -387,9 +387,56 @@ def get_zip_download_link(zip_file):
     href = f'<a href="data:application/zip;base64,{b64}" download="{zip_file}">Download All</a>'
     return href
 
+
+API_URL = f'https://tonpixzfvq3791u9.us-east-1.aws.endpoints.huggingface.cloud'
+headers = {
+	"Authorization": "Bearer XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX",
+	"Content-Type": "audio/wav"
+}
+
+def query(filename):
+    with open(filename, "rb") as f:
+        data = f.read()
+    response = requests.post(API_URL, headers=headers, data=data)
+    return response.json()
+
+def generate_filename(prompt, file_type):
+    central = pytz.timezone('US/Central')
+    safe_date_time = datetime.now(central).strftime("%m%d_%H%M")
+    replaced_prompt = prompt.replace(" ", "_").replace("\n", "_")
+    safe_prompt = "".join(x for x in replaced_prompt if x.isalnum() or x == "_")[:90]
+    return f"{safe_date_time}_{safe_prompt}.{file_type}"
+
+# 10. Audio recorder to Wav file:
+def save_and_play_audio(audio_recorder):
+    audio_bytes = audio_recorder()
+    if audio_bytes:
+        filename = generate_filename("Recording", "wav")
+        with open(filename, 'wb') as f:
+            f.write(audio_bytes)
+        st.audio(audio_bytes, format="audio/wav")
+        return filename
+
+# 9B. Speech transcription to file output - OPENAI Whisper
+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.")
+
+    # Audio, transcribe, GPT:
+    filename = save_and_play_audio(audio_recorder)
+    if filename is not None:
+        transcription = transcribe_audio(filename)
+        st.write(transcription)
+
+
 def main():
 
-    st.title("DromeLlama7B")
+    st.title("AI Drome Llama")
     prompt = f"Write ten funny jokes that are tweet length stories that make you laugh.  Show as markdown outline with emojis for each."
 
     # Add Wit and Humor buttons
@@ -402,16 +449,17 @@ def main():
         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).')
 
-    
     openai.api_key = os.getenv('OPENAI_KEY')
     menu = ["txt", "htm", "xlsx", "csv", "md", "py"]
     choice = st.sidebar.selectbox("Output File Type:", menu)
     model_choice = st.sidebar.radio("Select Model:", ('gpt-3.5-turbo', 'gpt-3.5-turbo-0301'))
-    filename = save_and_play_audio(audio_recorder)
-    if filename is not None:
-        transcription = transcribe_audio(key, filename, "whisper-1")
-        st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
-        filename = None
+    
+    #filename = save_and_play_audio(audio_recorder)
+    #if filename is not None:
+    #    transcription = transcribe_audio(key, filename, "whisper-1")
+    #    st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
+    #    filename = None
+        
     user_prompt = st.text_area("Enter prompts, instructions & questions:", '', height=100)
     collength, colupload = st.columns([2,3])  # adjust the ratio as needed
     with collength:
@@ -503,35 +551,35 @@ def main():
             create_file(filename, user_prompt, response, should_save)
             st.experimental_rerun()
 
-
     # Feedback
     # Step: Give User a Way to Upvote or Downvote
     feedback = st.radio("Step 8: Give your feedback", ("👍 Upvote", "👎 Downvote"))
-
     if feedback == "👍 Upvote":
         st.write("You upvoted 👍. Thank you for your feedback!")
     else:
         st.write("You downvoted 👎. Thank you for your feedback!")
-
-load_dotenv()
-st.write(css, unsafe_allow_html=True)
-st.header("Chat with documents :books:")
-user_question = st.text_input("Ask a question about your documents:")
-if user_question:
-    process_user_input(user_question)
-with st.sidebar:
-    st.subheader("Your documents")
-    docs = st.file_uploader("import documents", accept_multiple_files=True)
-    with st.spinner("Processing"):
-        raw = pdf2txt(docs)
-        if len(raw) > 0:
-            length = str(len(raw))
-            text_chunks = txt2chunks(raw)
-            vectorstore = vector_store(text_chunks)
-            st.session_state.conversation = get_chain(vectorstore)
-            st.markdown('# AI Search Index of Length:' + length + ' Created.')  # add timing
-            filename = generate_filename(raw, 'txt')
-            create_file(filename, raw, '', should_save)
+        
+    load_dotenv()
+    st.write(css, unsafe_allow_html=True)
+    st.header("Chat with documents :books:")
+    user_question = st.text_input("Ask a question about your documents:")
+    if user_question:
+        process_user_input(user_question)
+    with st.sidebar:
+        st.subheader("Your documents")
+        docs = st.file_uploader("import documents", accept_multiple_files=True)
+        with st.spinner("Processing"):
+            raw = pdf2txt(docs)
+            if len(raw) > 0:
+                length = str(len(raw))
+                text_chunks = txt2chunks(raw)
+                vectorstore = vector_store(text_chunks)
+                st.session_state.conversation = get_chain(vectorstore)
+                st.markdown('# AI Search Index of Length:' + length + ' Created.')  # add timing
+                filename = generate_filename(raw, 'txt')
+                create_file(filename, raw, '', should_save)
 
 if __name__ == "__main__":
-    main()
+    whisper_main()
+    main()
+