Update app.py

app.py

@@ -13,7 +13,6 @@ if "chat_history" not in st.session_state:
 def render_chat_history():
     """Render the current chat history in a dedicated container."""
     chat_container = st.container()
-    # Build the chat history markdown string.
     chat_md = ""
     for msg in st.session_state.chat_history:
         if msg["role"] == "user":
@@ -22,24 +21,24 @@ def render_chat_history():
             chat_md += f"**Assistant:** {msg['content']}\n\n"
     chat_container.markdown(chat_md)
 
-# =====================
-# Initial Conversation Generation
-# =====================
-
 st.title("Conversation Script Generator Using Groq API")
 
-# Input fields for initial generation.
-col1, col2 = st.columns(2)
+# ---------------------------
+# Initial Conversation Generation Inputs
+# ---------------------------
+col1, col2, col3 = st.columns(3)
 with col1:
     left_speaker = st.text_input("Left Speaker", placeholder="Enter left speaker name")
 with col2:
     right_speaker = st.text_input("Right Speaker", placeholder="Enter right speaker name")
+with col3:
+    contact_name = st.text_input("Contact Name", placeholder="Enter contact name for conversation")
 
 theme = st.text_area("Overall Theme of the Conversation", height=100)
 additional_details = st.text_area("Additional Conversation Details (Optional)", height=100)
 
 # Check if all required fields are filled.
-is_ready = bool(theme and left_speaker and right_speaker)
+is_ready = bool(theme and left_speaker and right_speaker and contact_name)
 
 if st.button("Generate Conversation", disabled=not is_ready):
     # Build the initial prompt.
@@ -113,7 +112,7 @@ Every script must begin with an insane, wacky brain rot hook that immediately gr
 "SIR WHY IS MY LAWYER ALSO MY UBER DRIVER?!" 👨‍⚖🚗💨
 Must be over-the-top and chaotic.
 Encourages immediate engagement.
-7️- Escalation & Progression
+7️⃣ Escalation & Progression
 The story must escalate into chaos but stay coherent:
 Start Crazy → Get Crazier → End With a Punchline or Twist
 Use sudden twists and reveals to maintain engagement.
@@ -132,14 +131,15 @@ The final line should be either a funny resolution or an abrupt exit.
 DO NOT USE THESE NAMES IN CONTACT OR IN TEXTS, THESE ARE ONLY TO TELL THE PROGRAM WHICH VOICE TO USE
    Left Actor = {left_speaker}
    Right Actor = {right_speaker}
+   Contact Name = {contact_name}
    Theme = {theme}
     """
     if additional_details.strip():
         prompt += f"\nAdditional Details: {additional_details}\n"
-
+    
     # Start the conversation history with the initial prompt.
     st.session_state.chat_history = [{"role": "user", "content": prompt}]
-
+    
     model = "llama-3.3-70b-versatile"
     try:
         chat_completion = client.chat.completions.create(
@@ -149,36 +149,46 @@ DO NOT USE THESE NAMES IN CONTACT OR IN TEXTS, THESE ARE ONLY TO TELL THE PROGRA
             max_completion_tokens=21890,
         )
         result_text = chat_completion.choices[0].message.content
-
+        
         if not result_text:
             st.error("The API call did not return any content.")
         else:
             st.success("Conversation generated successfully!")
-            # Append the generated conversation to chat history.
             st.session_state.chat_history.append({"role": "assistant", "content": result_text})
     except Exception as e:
         st.error(f"An error occurred while calling the API: {e}")
 
-# Render the current conversation.
 st.markdown("### Generated Conversation Script")
 render_chat_history()
 
-# =====================
-# Chat Interface for Modifications
-# =====================
-
+# ---------------------------
+# Chat Interface for Revisions
+# ---------------------------
 st.markdown("---")
 st.header("Chat & Modify Conversation Script")
 
 with st.form(key="chat_form", clear_on_submit=True):
-    user_message = st.text_area("Enter your message (e.g., request modifications)", height=100)
+    user_message = st.text_area("Enter your revision request (e.g., modify the script)", height=100)
     submit_chat = st.form_submit_button("Send")
 
 if submit_chat and user_message.strip():
-    # Append the user's message to chat history.
-    st.session_state.chat_history.append({"role": "user", "content": user_message})
+    # Retrieve the last assistant-generated script for context.
+    last_script = ""
+    for msg in reversed(st.session_state.chat_history):
+        if msg["role"] == "assistant":
+            last_script = msg["content"]
+            break
+    # Build a default revision prompt that includes context and instructions.
+    revision_prompt = f"""Revision Request: {user_message}
+
+Context: The current conversation script is as follows:
 
-    # Call the API with the updated conversation history.
+{last_script}
+
+Please revise the script to incorporate the above requested changes. Do not simply repeat the same content—update it with new, modified details that reflect the user's instructions.
+"""
+    st.session_state.chat_history.append({"role": "user", "content": revision_prompt})
+    
     model = "llama-3.3-70b-versatile"
     try:
         chat_completion = client.chat.completions.create(
@@ -193,21 +203,18 @@ if submit_chat and user_message.strip():
     except Exception as e:
         st.error(f"An error occurred while calling the API: {e}")
 
-# Render the updated conversation after processing the new message.
 st.markdown("### Updated Conversation Script")
 render_chat_history()
 
-# -------------------------
-# Download Option for the Latest Script (from the assistant response)
-# -------------------------
+# ---------------------------
+# Download Option for the Latest Assistant Script
+# ---------------------------
 if st.session_state.chat_history:
-    # Extract the last assistant response.
     last_response = ""
     for msg in reversed(st.session_state.chat_history):
         if msg["role"] == "assistant":
             last_response = msg["content"]
             break
-
     if last_response:
         txt_bytes = last_response.encode("utf-8")
         txt_io = io.BytesIO(txt_bytes)