Update app.py

app.py

@@ -8,38 +8,42 @@ client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
 
 # Initialize session state for chat history if it doesn't exist.
 if "chat_history" not in st.session_state:
-    st.session_state.chat_history = []  # Each entry: {"role": "user"/"assistant", "content": "..."} 
+    st.session_state.chat_history = []  # List of dicts: {"role": "user"/"assistant", "content": "..."}
+
+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":
+            chat_md += f"**User:** {msg['content']}\n\n"
+        else:
+            chat_md += f"**Assistant:** {msg['content']}\n\n"
+    chat_container.markdown(chat_md)
+
+# =====================
+# Initial Conversation Generation
+# =====================
 
-# Speaker options (unused in chat but kept for reference)
-speaker_options = {
-    "Liam": "A male voice",
-    "Dorothy": "A female voice",
-}
-
-# Build the Streamlit UI header.
 st.title("Conversation Script Generator Using Groq API")
 
-# Create columns for left/right speaker inputs.
+# Input fields for initial generation.
 col1, col2 = st.columns(2)
 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")
 
-# Text area for overall conversation theme.
 theme = st.text_area("Overall Theme of the Conversation", height=100)
-
-# Optional additional details.
 additional_details = st.text_area("Additional Conversation Details (Optional)", height=100)
 
-# Determine if required fields are filled.
+# Check if all required fields are filled.
 is_ready = bool(theme and left_speaker and right_speaker)
 
-# Container for initial conversation generation.
-with st.container():
-    if st.button("Generate Conversation", disabled=not is_ready):
-        # Build the initial prompt.
-        prompt = f"""
+if st.button("Generate Conversation", disabled=not is_ready):
+    # Build the initial prompt.
+    prompt = f"""
 📌 Objective
 Generate a highly engaging and comedic iMessage chat script formatted for video generation.
 The script follows a structured left-right chat format, maintains a meme-heavy, brain rot style, and incorporates sound effects, reactions, and the Cantina Roast Bot segment in a specific format.
@@ -129,75 +133,54 @@ DO NOT USE THESE NAMES IN CONTACT OR IN TEXTS, THESE ARE ONLY TO TELL THE PROGRA
    Left Actor = {left_speaker}
    Right Actor = {right_speaker}
    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(
-                messages=st.session_state.chat_history,
-                model=model,
-                temperature=1.2,
-                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})
-
-                # Display the generated conversation.
-                st.write("### Generated Conversation Script")
-                st.text_area("", result_text, height=300)
-
-                # Offer a download of the script.
-                txt_bytes = result_text.encode("utf-8")
-                txt_io = io.BytesIO(txt_bytes)
-                st.download_button(
-                    label="Download Script as TXT",
-                    data=txt_io,
-                    file_name="conversation_script.txt",
-                    mime="text/plain"
-                )
-        except Exception as e:
-            st.error(f"An error occurred while calling the API: {e}")
-
-# ------------------------------------------------------------------
-# Chat Interface Section
-# This section will appear after the initial generation (or if history already exists)
+    """
+    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(
+            messages=st.session_state.chat_history,
+            model=model,
+            temperature=1.2,
+            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
+# =====================
+
 st.markdown("---")
 st.header("Chat & Modify Conversation Script")
 
-# Display the chat history.
-for msg in st.session_state.chat_history:
-    if msg["role"] == "user":
-        st.markdown(f"**User:** {msg['content']}")
-    else:
-        st.markdown(f"**Assistant:** {msg['content']}")
-
-# Form to send a new chat message.
 with st.form(key="chat_form", clear_on_submit=True):
     user_message = st.text_area("Enter your message (e.g., request modifications)", height=100)
     submit_chat = st.form_submit_button("Send")
-    
-# ... (previous code remains unchanged)
 
 if submit_chat and user_message.strip():
-    # Append the new user message to chat history.
+    # Append the user's message to chat history.
     st.session_state.chat_history.append({"role": "user", "content": user_message})
-    
-    # Call the Groq API with the entire conversation history to keep memory intact.
+
+    # Call the API with the updated conversation history.
+    model = "llama-3.3-70b-versatile"
     try:
-        model = "llama-3.3-70b-versatile"
         chat_completion = client.chat.completions.create(
             messages=st.session_state.chat_history,
             model=model,
@@ -210,8 +193,27 @@ if submit_chat and user_message.strip():
     except Exception as e:
         st.error(f"An error occurred while calling the API: {e}")
 
-    # Conditionally rerun if available, otherwise inform the user.
-    if hasattr(st, "experimental_rerun"):
-        st.experimental_rerun()
-    else:
-        st.info("Please refresh the page to see the updated conversation.")
+# 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)
+# -------------------------
+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)
+        st.download_button(
+            label="Download Latest Assistant Script as TXT",
+            data=txt_io,
+            file_name="conversation_script.txt",
+            mime="text/plain"
+        )