Update app.py

app.py

@@ -23,8 +23,6 @@ menu = ["txt", "htm", "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'))
 
-uploaded_files = []
-
 def generate_filename(prompt, file_type):
     central = pytz.timezone('US/Central')
     safe_date_time = datetime.now(central).strftime("%m%d_%I%M")  
@@ -65,62 +63,60 @@ def save_and_play_audio(audio_recorder):
         with open(filename, 'wb') as f:
             f.write(audio_bytes)
         st.audio(audio_bytes, format="audio/wav")
-        uploaded_files.append(filename)  # Add the new file name to the list
         return filename
     return None
 
-# ... (All your other function definitions)
-
-def main():
-    user_prompt = st.text_area("Enter prompts, instructions & questions:", '', height=100)
-
-    collength, colupload = st.columns([2,3])  # adjust the ratio as needed
-    with collength:
-        max_length = st.slider("File section length for large files", min_value=1000, max_value=128000, value=12000, step=1000)
-    with colupload:
-        uploaded_file = st.file_uploader("Add a file for context:", type=["xml", "json", "html", "htm", "md", "txt"])
-    
-    document_sections = deque()
-    document_responses = {}
+def transcribe_and_chat(openai_key, file_path, model):
+    transcription = transcribe_audio(openai_key, file_path, model)
+    if transcription is not None:
+        response = chat_with_model(transcription, '')
+        return transcription, response
+    else:
+        return None, None
 
-    if uploaded_file is not None:
-        file_content = uploaded_file.getvalue().decode('utf-8')
-        # Handle different file types here
-        # ...
-        document_sections.append(file_content)
-        
-    new_filename = save_and_play_audio(audio_recorder)
-    if new_filename is not None:
-        st.write(f'File {new_filename} uploaded.')
-        if st.button("Transcribe"):
-            transcription = transcribe_audio(openai.api_key, new_filename, "whisper-1")
-            st.write(transcription)
-            chat_with_model(transcription, '') # push transcript through as prompt
+def create_file(filename, prompt, response):
+    if filename.endswith(".txt"):
+        with open(filename, 'w') as file:
+            file.write(f"Prompt:\n{prompt}\nResponse:\n{response}")
+    elif filename.endswith(".htm"):
+        with open(filename, 'w') as file:
+            file.write(f"<h1>Prompt:</h1> <p>{prompt}</p> <h1>Response:</h1> <p>{response}</p>")
+    elif filename.endswith(".md"):
+        with open(filename, 'w') as file:
+            file.write(f"# Prompt: \n {prompt} \n # Response: \n {response}")
+    elif filename.endswith(".py"):
+        with open(filename, 'w') as file:
+            file.write(f"# Prompt: \n'''{prompt}'''\n # Response: \n'''{response}'''")
+    else:
+        st.error("Unsupported file type!")
 
-    if st.button('💬 Chat'):
-        user_responses = document_responses.get(user_prompt, [])
-        if len(user_responses) > 0:
-            st.write(user_responses[-1])
-        else:
-            if document_sections:
-                st.write('First document section:')
-                st.write(document_sections[0])
-                document_responses[user_prompt] = [chat_with_model(user_prompt, document_sections[0])]
-                st.write(document_responses[user_prompt][-1])
-            else:
-                document_responses[user_prompt] = [chat_with_model(user_prompt, '')]
-                st.write(document_responses[user_prompt][-1])
-                
-    if uploaded_files:
-        st.write(f'Last uploaded file: {uploaded_files[-1]}')
+def main():
+    st.sidebar.header("Choose your Input Method:")
+    input_choice = st.sidebar.radio("",('Type it out','Record it'))
 
-    for filename in uploaded_files:
-        if st.button(f"Transcribe and Chat for {filename}"):
-            transcription, response = transcribe_and_chat(openai.api_key, filename, "whisper-1")
-            if transcription is not None and response is not None:
-                filename = generate_filename(transcription, choice)
-                create_file(filename, transcription, response)
-                st.sidebar.markdown(get_table_download_link(filename), unsafe_allow_html=True)
+    if input_choice == 'Type it out':
+        st.header("Type your query")
+        prompt = st.text_area("Input")
+        document_section = st.text_area("Document Section")
+        if st.button("Get Response"):
+            filename = generate_filename(prompt, choice)
+            response = chat_with_model(prompt, document_section)
+            st.write('Response:')
+            st.write(response)
+            create_file(filename, prompt, response)
+    elif input_choice == 'Record it':
+        st.header("Record your query")
+        st.write("Press record to start recording")
+        audio_recorder = audio_recorder()
+        filename = save_and_play_audio(audio_recorder)
+        if filename:
+            if st.button("Transcribe and Chat"):
+                transcription, response = transcribe_and_chat(openai.api_key, filename, "whisper-1")
+                if transcription and response:
+                    st.write('Transcription:')
+                    st.write(transcription)
+                    st.write('Response:')
+                    st.write(response)
 
 if __name__ == "__main__":
     main()