Update app.py

app.py

@@ -6,55 +6,51 @@ from groq import Groq
 # Set up Groq client using the environment variable for API key.
 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": "..."} 
 
-# Speaker options
+# Speaker options (unused in chat but kept for reference)
 speaker_options = {
     "Liam": "A male voice",
     "Dorothy": "A female voice",
 }
 
-# Build the Streamlit UI
+# Build the Streamlit UI header.
 st.title("Conversation Script Generator Using Groq API")
 
+# Create columns for left/right speaker inputs.
 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
+# Text area for overall conversation theme.
 theme = st.text_area("Overall Theme of the Conversation", height=100)
 
-# Optional additional details
+# Optional additional details.
 additional_details = st.text_area("Additional Conversation Details (Optional)", height=100)
 
-# Check if all required fields are filled
+# Determine if required fields are filled.
 is_ready = bool(theme and left_speaker and right_speaker)
 
-# Disable the button if required fields are not filled
-if st.button("Generate Conversation", disabled=not is_ready):
-
-    # Build the prompt message for Groq API.
-    prompt = (
-        f"""
-        📌 Objective
+# Container for initial conversation generation.
+with st.container():
+    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.
-
 📝 Script Structure & Formatting Rules
 1️- Contact Name (Title)
 At the beginning of the script, define the contact name, appearing at the top of the iMessage screen.
-
 <Contact Name>:<Contact Name>
 ✅ Example:
-
-
 Amir (expensive af delivery guy🍆💸):Amir (expensive af delivery guy🍆💸)
 2️- Dialogue Format
 Each dialogue line follows this format:
-
 <Direction>: <SpeakerName> > <Phrase 1> <Phrase 2> <Phrase 3> <Phrase 4> [Reaction]
 Left: Represents the left speaker (User).
 Right: Represents the right speaker (Other character).
@@ -62,52 +58,36 @@ SpeakerName: A randomly assigned name, but characters do not know their own name
 Phrase Limit: Max 4 phrases per line.
 Reaction Cues: Placed at the end of lines, enclosed in square brackets [].
 ✅ Example:
-
-
 Left: Unknown Guy> Bro what is this? This is not food [shocked]
-Right: Amir > My guy that's a gourmet dish 🤨 [suspense_music]
+Right: Amir> My guy that's a gourmet dish 🤨 [suspense_music]
 Left: Unknown Guy> Gourmet?? It’s literally burnt [explosion]
-Right: Amir > Charcoal adds flavor, trust me bro 😎
+Right: Amir> Charcoal adds flavor, trust me bro 😎
 3️- Bracketed Cues & Sound Effects
 Use brackets [ ] for sound effects, reactions, or video cues.
-
 [shocked] → Must be used often
 [fart2] → Should be used less
 [fortnite] → Should be used less
 [get_out] → Must ONLY be used at the end of the conversation
 ✅ Example:
-
-sql
-Copy
-Edit
 Right: Amir> BRO WHY IS THERE A RAT IN MY BURGER?!! [shocked] 🐀🍔
 Left: Unknown Guy> Extra protein sir 😎
 Right: Amir> 💀💀💀 [fortnite]
 4️- Image Placeholders
 For inserting images, use:
-
-php-template
-Copy
-Edit
 <Direction>: img:<DescriptiveImageName>
 ✅ Example:
-
-
 Left: img:burnt_food_delivery
 Right: Amir> BRO WHY IS MY PIZZA LOOKING LIKE CHARCOAL?!! [shocked]
 Min. 3 images per script.
 Descriptive placeholders must be clear.
 5️- Cantina App (Roast Bot Segment)
 🚨 Correct Format (ALWAYS THE SAME STRUCTURE)
-
 Right speaker initiates by opening Cantina.
 Right speaker asks Roast Bot for help.
 Left speaker (ALWAYS Antoni) responds.
 Right speaker must repeat what Roast Bot told them to say.
 Contact Name Title must reappear after Cantina.
 ✅ Correct Example:
-
-
 Cantina: Roast bot:Roast bot pfp == Alex> Let me open the cantina app to talk to Roast bot
 Right: Alex> Hey Roast Bot  
 Right: Alex> Amir is mad at me for not wanting to tip him  
@@ -118,16 +98,12 @@ Left: Antoni> maybe you should start working for real and not begging for tips a
 Amir (expensive af delivery guy🍆💸):Amir (expensive af delivery guy🍆💸)  
 Right: Alex> maybe you should start working for real and not begging for tips all the time  
 🔹 Rules:
-
 Right speaker MUST repeat the roast word-for-word.
 The contact name/title must reappear after the Cantina segment.
 DO NOT GIVE HEADING TO SEGMENTS, all the script should be in one flow and should be at least 30 messages long
 6️⃣ Brain Rot Hook (MANDATORY)
 Every script must begin with an insane, wacky brain rot hook that immediately grabs attention.
-
 ✅ Examples:
-
-=
 "🚨 BRO I JUST ORDERED UBER EATS AND THIS GUY SHOWED UP IN A SPACESHIP???" 🚀👀
 "WHY DID MY DAD JUST SAY HE WANTS TO PLAY GTA WITH ME???" 🎮💀
 "SIR WHY IS MY LAWYER ALSO MY UBER DRIVER?!" 👨‍⚖🚗💨
@@ -135,12 +111,10 @@ Must be over-the-top and chaotic.
 Encourages immediate engagement.
 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.
 The final line should be either a funny resolution or an abrupt exit.
 ✅ Example Progression:
-
 🚀 Crazy Hook:
 "BRO WHY IS MY PIZZA DELIVERY GUY ALSO MY LAWYER???" 🍕👨‍⚖
 🔥 Escalation:
@@ -152,60 +126,87 @@ The final line should be either a funny resolution or an abrupt exit.
 🏃‍♂️ Dramatic Exit (ALWAYS USE [get_out])
 "Left: Lawyer> I quit. [get_out]"
 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}
-    Theme={theme}     
+   Left Actor = {left_speaker}
+   Right Actor = {right_speaker}
+   Theme = {theme}
         """
-    )
-    if additional_details.strip():
-        prompt += f"Additional Details: {additional_details}\n"
+        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"
 
-    # Prepare the message payload for the Groq API call.
-    messages = [
-        {
-            "role": "user",
-            "content": prompt,
-        }
-    ]
+        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)
+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")
     
-    # Model selection: adjust as needed.
+if submit_chat and user_message.strip():
+    # Append the new user message to chat history.
+    st.session_state.chat_history.append({"role": "user", "content": user_message})
     
-    model = "llama-3.3-70b-versatile"
-
+    # Call the Groq API with the entire conversation history to keep memory intact.
     try:
-        # Make the chat completion call using Groq.
+        model = "llama-3.3-70b-versatile"
         chat_completion = client.chat.completions.create(
-            messages=messages,
+            messages=st.session_state.chat_history,
             model=model,
             temperature=1.2,
             max_completion_tokens=21890,
-
-            
         )
-
-        # Extract the generated conversation from the API response.
-        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!")
-            
-            # Display the result inside a text area.
-            st.write("### Generated Conversation Script")
-            st.text_area("", result_text, height=300)
-            
-            # Create a downloadable TXT file.
-            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"
-            )
-
+        reply = chat_completion.choices[0].message.content
+        st.session_state.chat_history.append({"role": "assistant", "content": reply})
+        st.success("Response received!")
     except Exception as e:
-        st.error(f"An error occurred while calling the API: {e}")
+        st.error(f"An error occurred while calling the API: {e}")
+
+    # Refresh the chat display.
+    st.experimental_rerun()