Update app.py

app.py

@@ -3,83 +3,100 @@ import os
 from openai import OpenAI
 from datetime import datetime
 
-# Initialize OpenAI API client 
-client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
+# Get dateTime strubg to build a filename reflecting the UserID + Timestamp
+dt = datetime.now()
+dt_string = str(dt)
 
-# Function to generate speech
-def generate_speech(name, raw_input_text):
+# Initialize OpenAI API client with API key
+client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
 
-    # Capture input text
-    input_text = raw_input_text  
+# Initialize empty history list to maintain the conversation context
+history_openai_format = [
+    {"role": "system", "content": (os.getenv("PROMPT"))}
+]
 
-    # Make history local
-    history_openai_format = [
-        {"role": "system", "content": (os.getenv("PROMPT"))}
-    ]
+# Define the function that takes a name and text input to generate the speech
+def generate_speech(name, raw_input_text):
+    input_text = raw_input_text
+    global history_openai_format  # Use the global history variable to maintain state
 
-    # Get dateTime string  
+    # Get dateTime string to build a filename reflecting the UserID + Timestamp
     dt = datetime.now()
     dt_string = str(dt)
 
-    # Define user ID and file
-    user_id = name if name else "unidentified"  
-    user_hist_file = "jcTSS-" + user_id + ".txt"
-
-    # Check if user history file exists
-if os.path.exists(user_hist_file):
-    with open(user_hist_file, "r", encoding="UTF-8") as file:
-        user_hist = file.read().strip()  
-else:
-    with open(user_hist_file, "w", encoding="UTF-8") as file:
-        file.write("User ID: " + user_id)
-
-# Next lines no longer indented too far left  
-input_text1 = f"I'm {name}. " + input_text
-history_openai_format.append({"role": "user", "content": input_text1})
-
-    # Rest of function
-
-    with open(user_hist_file, "a+", encoding="UTF-8") as file:
-        file.write("\n\nDate/Time: " + datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
-        for message in history_openai_format[-2:]:
-            file.write(f"\n{message['role'].title()}: {message['content']}")
+    # Define the user ID and construct the filename for the user's history file
+    user_id = name if name else "unidentified"  # User identification, typically an email.
+    user_hist_file = "jcTSS-" + user_id + ".txt"  # Filename where the user history will be stored.
+
+    # Check if the user's history file exists
+    if os.path.exists(user_hist_file):
+        # If it exists, open and read its contents, then print it
+        with open(user_hist_file, "r", encoding="UTF-8") as file:
+            user_hist = file.read().strip()  # Remove leading/trailing whitespace
+    else:
+        # If it does not exist, create the file and initialize it with the user ID
+        with open(user_hist_file, "w", encoding="UTF-8") as file:
+            file.write("User ID: " + user_id)
+    
+    # Append user message to history with the name included
+    input_text1 = f"I'm {name}. " + input_text
+    history_openai_format.append({"role": "user", "content": input_text1})
 
-    # Get OpenAI response
+    # Build completion with OpenAI using the accumulated history
     completion = client.chat.completions.create(
-         model="gpt-3.5-turbo-1106",
-         messages=history_openai_format
+        model="gpt-3.5-turbo-1106",
+        messages=history_openai_format
     )
 
+    # Extract generated text (response by the assistant) from completion
     message_content = completion.choices[0].message.content.strip()
-    
-    # Remove "Johnny" prefix
+
+    # Remove "Johnny" from the beginning of the assistant's message if present
     if message_content.lower().startswith("johnny"):
-         message_content = message_content[6:].strip()
+        # Strip the leading "Johnny" from the message content
+        message_content = message_content[6:].strip()
 
-    # Append assistant response 
+    # Append assistant's message to history
     history_openai_format.append({"role": "assistant", "content": message_content})
 
-    # Convert text to speech
+    # Use OpenAI's text-to-speech API to convert the text response to audio
     response = client.audio.speech.create(
-         model="tts-1",
-         voice="onyx",
-         input=str(message_content)
+        model="tts-1",
+        voice="onyx",
+        input=str(message_content)
     )
 
-    # Generate transcript
+    # Prepare the transcript for the TextArea output
+    # Exclude the system message from the transcript and show the user's actual input
+    # Add line spaces between user and assistant output, and between user/assistant exchanges
     transcript = ""
     for i, msg in enumerate(history_openai_format):
         if msg['role'] != 'system':
-             # Append user and assistant messages
-    
-    # Write user history file  
+            if msg['role'] == 'user':
+                transcript += "Date/Time: " + datetime.now().strftime("%Y-%m-%d %H:%M:%S") + "\n"
+                transcript += f"Guest: {input_text}\n"  # Change 'User' to 'Guest'
+            else:
+                transcript += f"Johnny: {msg['content']}\n\n"  # Change 'Assistant' to 'Johnny'
+
+    # Write the user and assistant messages to the history file after the exchange
     with open(user_hist_file, "a+", encoding="UTF-8") as file:
         file.write("\n\nDate/Time: " + datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
-        for message in history_openai_format[-2:]:  
+        for message in history_openai_format[-2:]:  # Last 2 messages include the user and assistant responses
             file.write(f"\n{message['role'].title()}: {message['content']}")
 
+    # Return the binary audio data and the transcript
     return response.content, transcript
 
-# Launch interface
-iface = gr.Interface(...)   
-iface.launch()
+# Define the Gradio interface with inputs for name and user text
+iface = gr.Interface(
+    fn=generate_speech,
+    inputs=[
+        gr.Textbox(label="Your Name (REQUIRED):", placeholder="Enter your FIRST NAME"),
+        gr.Textbox(label="Your question or comment for Johnny:")
+    ],
+    outputs=[gr.Audio(autoplay=True, label="Johnny's response:"), gr.TextArea(label="Transcript", autoscroll="True", show_copy_button="True")],
+    live=False
+)
+
+# Launch the interface
+iface.launch(show_api=False)