Update app.py

app.py

@@ -8,17 +8,15 @@ 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 = []  # List of dicts: {"role": "user"/"assistant", "content": "..."}
+    st.session_state.chat_history = []  # Each item is a dict: {"role": "user"/"assistant", "content": "..."}
 
-def render_chat_history():
-    """Render the current chat history in a dedicated container."""
+def render_assistant_history():
+    """Render only the assistant responses from the conversation history."""
     chat_container = st.container()
     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"
+        if msg["role"] == "assistant":
+            chat_md += f"**Assistant:**\n{msg['content']}\n\n"
     chat_container.markdown(chat_md)
 
 st.title("Conversation Script Generator Using Groq API")
@@ -41,21 +39,17 @@ additional_details = st.text_area("Additional Conversation Details (Optional)",
 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.
     prompt = f"""
-
-
- Script Structure & Formatting Rules
+    Youe Task is to  generate an engaging script which is a conversation between two people on iMessage for meme purposes.. It should strictly follow a given 
+    syntax and should be diverse and creative
+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:
-
 Liam (Black Market Electronics Dealer 📱💸):Liam (Black Market Electronics Dealer 📱💸)
 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).
@@ -63,45 +57,37 @@ SpeakerName: A randomly assigned alias (characters do not know their own names).
 Phrase Limit: Max 4 phrases per line.
 Reaction Cues: Placed at the end of lines, enclosed in square brackets [].
 ✅ Example:
-
 Left: MysteryGuy> BRO WHY IS MY COMPUTER SPEAKING LATIN??? [shocked]
 Right: Liam> Uh… you *might* have summoned something 💀 [suspense_music]
 Left: MysteryGuy> SUMMONED WHAT???
 Right: Liam> 😶
 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:
-
-
 Right: Max> BRO WHY IS THERE A ROBOT FOLLOWING ME?!! [shocked] 🤖
 Left: UnknownDude> Congrats, you just activated *follow mode* 😎
 Right: Max> FOLLOW MODE??? TURN IT OFF
 Left: UnknownDude> nahhh that's a premium feature 🤣💀 [fortnite]
 4️ Image Placeholders
 For inserting images, use:
-
 <Direction>: img:<DescriptiveImageName>
 ✅ Example:
-
 Left: img:ancient_phone_booting_up
 Right: Liam> WHY DID MY OLD NOKIA JUST START SCREAMING??? [shocked]
 Min. 3 images per script.
 Descriptive placeholders must be varied and 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> Liam is mad at me for borrowing his Tesla without permission  
@@ -114,15 +100,12 @@ Right: Alex> maybe if you had better security, you wouldn’t have to track your
 🚫 Incorrect Example (DO NOT DO THIS):
 Cantina: Roast bot:Roast bot pfp == Liam> Hey Roast Bot, we need your help! My AI fridge is trying to blackmail me! ❌
 🔹 Rules:
-
 Left Speaker in Cantina is ALWAYS "Antoni."
 Right speaker MUST repeat the roast word-for-word.
 The contact name/title must reappear after the Cantina segment.
 6️ Brain Rot Hook (MANDATORY)
 Every script must begin with an insane, wacky brain rot hook that immediately grabs attention.
-
 ✅ Examples:
-
 "WHY DID MY TV JUST ASK ME FOR A PASSWORD???" 📺🔑  
 "BRO MY DOG JUST SENT ME A TEXT MESSAGE???" 🐶📲  
 "WHY IS MY LANDLORD SELLING MY HOUSE ON FACEBOOK MARKETPLACE???" 🏠💀  
@@ -130,12 +113,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 DID MY PHONE JUST ORDER A UBER ON ITS OWN???" 🚗💨
 🔥 Escalation:
@@ -148,12 +129,10 @@ The final line should be either a funny resolution or an abrupt exit.
 "Left: UnknownDude> Enjoy the ride 😎 [get_out]"
 8️⃣ Speaker Name Randomization
 🚨 Speakers should NOT know their own names.
-
 Left speaker’s name should always appear as "Unknown Guy" or a random alias.
 Right speaker’s name should also be replaced with something generic unless necessary.
 The system should NOT recognize them.
 ✅ Example:
-
 Left: MysteriousCaller> BRO WHO JUST HACKED MY PHONE??? [shocked]
 Right: HackerDude> Oh, don’t worry, it's just an update 😎
 Left: MysteriousCaller> UPDATE??? BRO WHAT DO YOU MEAN
@@ -164,7 +143,6 @@ Emoji Button → Clicking adds 5 random emojis to any phrase.
 Max 9 sound effect commands, with [shocked] used most.
 Final Parameters & Instructions
 When prompting the AI, specify:
-
 Left Actor = {left_speaker}
 Right Actor = {right_speaker}
 Theme = {theme} (e.g., “AI Gone Wrong”, “Haunted Electronics”, “Unexpected Legal Trouble”)
@@ -178,7 +156,6 @@ Speaker names are randomized and unknown to them.
 🔹 Cantina Roast Bot always follows the same structure.
 🔹 Brain rot hook is mandatory.
 🔹 Escalates into chaos but remains coherent.
-
 Contact Name: {contact_name}
 {contact_name} : {contact_name} in the start
 Left Speaker: {left_speaker}
@@ -204,7 +181,6 @@ for right ones
             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:
@@ -214,7 +190,7 @@ for right ones
         st.error(f"An error occurred while calling the API: {e}")
 
 st.markdown("### Generated Conversation Script")
-render_chat_history()
+render_assistant_history()
 
 # ---------------------------
 # Chat Interface for Revisions
@@ -233,14 +209,14 @@ if submit_chat and user_message.strip():
         if msg["role"] == "assistant":
             last_script = msg["content"]
             break
-    # Build a default revision prompt that includes context and instructions.
+    # Build a revision prompt including context and user request.
     revision_prompt = f"""Revision Request: {user_message}
 
 Context: The current conversation script is as follows:
 
 {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.
+Please revise the script to incorporate the requested changes. Do not simply repeat the same content—update it with new, modified details reflecting the user's instructions.
 """
     st.session_state.chat_history.append({"role": "user", "content": revision_prompt})
     
@@ -259,7 +235,7 @@ Please revise the script to incorporate the above requested changes. Do not simp
         st.error(f"An error occurred while calling the API: {e}")
 
 st.markdown("### Updated Conversation Script")
-render_chat_history()
+render_assistant_history()
 
 # ---------------------------
 # Download Option for the Latest Assistant Script