Deploy Run Buddy with sample chat

app.py

@@ -0,0 +1,174 @@
+import logging
+import os
+import random
+from openai import OpenAI
+from gtts import gTTS
+import gradio as gr
+
+# Configure logging
+logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+# Initialize OpenAI client with API key from environment variable
+try:
+    os.environ["OPENAI_API_KEY"] = os.getenv("OPENAI_API_KEY")
+    if not os.environ["OPENAI_API_KEY"]:
+        raise ValueError("OPENAI_API_KEY environment variable not set!")
+    client = OpenAI()
+except Exception as e:
+    logger.error(f"Failed to initialize OpenAI client: {e}")
+
+# Function to process chat and generate outputs
+def motivate_me(chat_file, name):
+    # Validate inputs
+    if not name:
+        logger.warning("No name provided")
+        return "Please enter your name!", "", "", None
+    if chat_file is None:
+        logger.warning("No chat file uploaded")
+        return "Please upload a WhatsApp chat file!", "", "", None
+
+    # Read random 500 lines from uploaded file
+    try:
+        with open(chat_file.name, "r", encoding="utf-8") as f:
+            lines = f.readlines()
+    except Exception as e:
+        logger.error(f"Error reading chat file: {e}")
+        return f"Error reading file: {str(e)}", "", "", None
+
+    if len(lines) >= 500:
+        start_index = random.randint(0, len(lines) - 500)
+        chunks = lines[start_index:start_index + 500]
+    else:
+        chunks = lines
+    logger.info(f"Selected random 500 lines, last 5: {chunks[-5:]}")
+
+    if not chunks:
+        logger.warning("No messages found in selected lines")
+        return "No messages found in the selected 500 lines!", "", "", None
+
+    messages_text = "".join(chunks)
+
+    # Summarize with OpenAI
+    summary_prompt = """Summarize the following WhatsApp conversation in 2–3 sentences, focusing only on notable events, emotional states, and meaningful interactions between 'You' ({name}) and the other user.
+
+    Highlight specific events involving {name}—especially any related to health, job, mental health, or finances.
+
+    If no major event is present, capture any noteworthy daily moments (e.g., cooking, commuting, funny moments).
+
+    Describe how {name} is feeling and what they are going through, using the format: {name} is feeling [emotion] and going through [event].
+
+    Always include the name of the other user {name} is talking to.
+    Exclude all sexual content.
+    Do not mention if "no major event" occurred—just describe what's there.
+
+    If a funny joke or banter is present, include that specific message in quotes for potential reuse.
+
+    Conversation:
+    {messages}"""
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": "You are a helpful assistant that summarizes concisely about how the user is feeling and what interactions are happening."},
+                {"role": "user", "content": summary_prompt.format(messages=messages_text, name=name)}
+            ],
+            max_tokens=500,
+            temperature=0.8
+        )
+        summary = response.choices[0].message.content.strip()
+        logger.info(f"Conversation Summary: {summary}")
+    except Exception as e:
+        logger.error(f"Error generating summary: {e}")
+        return f"Error summarizing chat: {str(e)}", "", "", None
+
+    # Extract a happy/exciting interaction
+    story_prompt = """Review the following WhatsApp conversation and extract **one clear, HAPPY & exciting interaction** between participants (e.g., playful plans like "can’t wait to see you" or "let’s cook together","you got this").
+
+    - Choose only **one uplifting moment** from the chat and summarize it in 5 words about why they said it.
+    - **Exclude all sexual or suggestive content.**
+    Conversation:  
+    {messages}"""
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": "You are a cute friend finding light-hearted, heartwarming, and positive messages."},
+                {"role": "user", "content": story_prompt.format(messages=messages_text)}
+            ],
+            max_tokens=100,
+            temperature=0.5
+        )
+        story = response.choices[0].message.content.strip()
+        logger.info(f"Processed Story: {story}")
+    except Exception as e:
+        logger.error(f"Error generating story: {e}")
+        return summary, f"Error finding story: {str(e)}", "", None
+
+    # Generate motivation with OpenAI
+    motivation_prompt = """Using this summary: '{summary}' and this interaction: '{story}', craft a short, personal, upbeat motivational message for {name} about running.
+
+    Highlight the details from the chat and mention the user name and the moments they shared.
+
+    Emphasize how the other user brings energy, joy, and small transformations to daily life and so does running.
+
+    You could also reference the included message exactly as it appears in the summary or story—remind them of it while tying it to running, if you wish to. Only add that message if it has meaning on its own. Dont add a single worded message.
+    Think about that message and then connect it with the situation in which they said it and top it up with a nice motivational quote!
+
+    Connect everything into something sweet, and encouraging.
+
+    If the summary has sad/worrisome emotion - you can use something soothing to acknowledge the hurt. Example: "It’s okay to feel heavy. Let your feet be light for now—run it off, breathe, come back brighter. ({name})" Come up with your own.
+
+    Keep in mind, all these incidents happened in the PAST. You are just reminding the user of their strength/fun loving nature.
+    Word limit: 50 words
+    Tone: Positive, personal, and warm
+    Must do: Mention the name of the other user in this motivational message.
+    End with: ({name})"""
+    try:
+        response = client.chat.completions.create(
+            model="gpt-4o-mini",
+            messages=[
+                {"role": "system", "content": "You are a motivational coach using personal events and stories from a user's life."},
+                {"role": "user", "content": motivation_prompt.format(summary=summary, story=story, name=name)}
+            ],
+            max_tokens=100,
+            temperature=0.7
+        )
+        motivation = response.choices[0].message.content.strip()
+        logger.info(f"Motivation generated: {motivation}")
+    except Exception as e:
+        logger.error(f"Error generating motivation: {e}")
+        return summary, story, f"Error generating motivation: {str(e)}", None
+
+    # Save and return audio
+    try:
+        tts = gTTS(motivation, lang="en")
+        audio_file = "motivation.mp3"
+        tts.save(audio_file)
+        logger.info("Audio file generated successfully")
+    except Exception as e:
+        logger.error(f"Error generating audio: {e}")
+        return summary, story, motivation, None
+
+    return summary, story, motivation, audio_file
+
+# Gradio interface
+interface = gr.Interface(
+    fn=motivate_me,
+    inputs=[
+        gr.File(label="Upload WhatsApp Chat (_chat.txt)"),
+        gr.Textbox(label="Your Name", value="", placeholder="Enter your name (e.g., Alex)")
+    ],
+    outputs=[
+        gr.Textbox(label="Conversation Summary"),
+        gr.Textbox(label="Wholesome Interaction"),
+        gr.Textbox(label="Your Motivation"),
+        gr.Audio(label="Listen Up!")
+    ],
+    title="Run Buddy: Your Personal Motivator",
+    description="Upload your WhatsApp chat (_chat.txt) and enter your name to get a summary, a happy moment, and a running pep talk from 500 random messages! Check out the sample file [here](https://huggingface.co/spaces/your-username/RunBuddyMotivator/raw/main/sample_chat.txt).",
+    theme="soft"
+)
+
+if __name__ == "__main__":
+    interface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+openai
+gtts
+gradio

sample_chat.txt

@@ -0,0 +1,100 @@
+[03/01/25, 09:00:00] Alex: Morning! How’s my favorite human?
+[03/01/25, 09:01:10] Jamie: Tired. Coffee machine broke again.
+[03/01/25, 09:02:20] Alex: You’re a zombie without coffee!
+[03/01/25, 09:03:30] Jamie: True. Shuffling to the store now.
+[03/01/25, 09:04:40] Alex: Don’t trip over your slippers!
+[03/01/25, 09:05:50] Jamie: Haha, "fell into a bush once!"
+[03/01/25, 09:06:00] Alex: Legendary. Bush still mad?
+[03/01/25, 09:07:10] Jamie: Probably plotting revenge.
+[03/01/25, 09:08:20] Alex: Watch out for ninja leaves!
+[03/01/25, 09:09:30] Jamie: I’ll wear armor next time!
+[03/01/25, 10:00:00] Alex: Work today?
+[03/01/25, 10:01:15] Jamie: Nope. Got laid off yesterday.
+[03/01/25, 10:02:25] Alex: What?! Are you okay?
+[03/01/25, 10:03:35] Jamie: Shocked. Feeling lost honestly.
+[03/01/25, 10:04:45] Alex: That sucks. Bosses are idiots.
+[03/01/25, 10:05:55] Jamie: Yeah, "budget cuts" they said.
+[03/01/25, 10:06:05] Alex: Budget cuts my foot!
+[03/01/25, 10:07:15] Jamie: Now I’m jobless and coffeeless.
+[03/01/25, 10:08:25] Alex: We’ll fix that. Coffee first?
+[03/01/25, 10:09:35] Jamie: Please. Meet at the café?
+[03/01/25, 11:00:00] Alex: Just spilled coffee on my shirt.
+[03/01/25, 11:01:10] Jamie: Haha, "you’re a walking disaster!"
+[03/01/25, 11:02:20] Alex: Only when you’re around!
+[03/01/25, 11:03:30] Jamie: Blaming me for your clumsiness?
+[03/01/25, 11:04:40] Alex: Yup, you’re my chaos magnet.
+[03/01/25, 11:05:50] Jamie: Fair. I attract chaos naturally.
+[03/01/25, 12:00:00] Alex: Lunch plans?
+[03/01/25, 12:01:15] Jamie: Cooking pasta. Wanna join?
+[03/01/25, 12:02:25] Alex: Only if we don’t burn it!
+[03/01/25, 12:03:35] Jamie: "Last time was epic flames!"
+[03/01/25, 12:04:45] Alex: Fire alarm’s still traumatized.
+[03/01/25, 12:05:55] Jamie: We’ll be careful this time.
+[03/01/25, 12:06:05] Alex: Famous last words!
+[03/01/25, 13:00:00] Jamie: Job hunt starts tomorrow.
+[03/01/25, 13:01:10] Alex: You got this, Jamie!
+[03/01/25, 13:02:20] Jamie: Hope so. Feeling down still.
+[03/01/25, 13:03:30] Alex: Let’s distract you tonight.
+[03/01/25, 13:04:40] Jamie: Movie night?
+[03/01/25, 13:05:50] Alex: Yup! Popcorn’s on me.
+[03/01/25, 14:00:00] Jamie: Just tripped over my cat.
+[03/01/25, 14:01:10] Alex: Cat okay? You okay?
+[03/01/25, 14:02:20] Jamie: Cat glared. I’m bruised.
+[03/01/25, 14:03:30] Alex: "Cat’s plotting world domination now!"
+[03/01/25, 14:04:40] Jamie: Probably. She’s the boss here.
+[03/01/25, 14:05:50] Alex: Bow to Queen Cat!
+[03/01/25, 15:00:00] Jamie: Applied to three jobs.
+[03/01/25, 15:01:10] Alex: Nice! Fingers crossed.
+[03/01/25, 15:02:20] Jamie: Thanks. Waiting’s the worst.
+[03/01/25, 15:03:30] Alex: Keep busy—run with me later?
+[03/01/25, 15:04:40] Jamie: Maybe. Legs feel like jelly.
+[03/01/25, 16:00:00] Alex: Bought groceries for dinner.
+[03/01/25, 16:01:10] Jamie: Sweet! What’s on the menu?
+[03/01/25, 16:02:20] Alex: Tacos. No fires this time!
+[03/01/25, 16:03:30] Jamie: "Taco Tuesday flashback—spicy chaos!"
+[03/01/25, 16:04:40] Alex: We survived the salsa explosion!
+[03/01/25, 16:05:50] Jamie: Barely. Kitchen smelled for days.
+[03/01/25, 17:00:00] Jamie: Cat stole a taco shell.
+[03/01/25, 17:01:10] Alex: Haha, "cat’s a taco bandit!"
+[03/01/25, 17:02:20] Jamie: She crunched it loudly.
+[03/01/25, 17:03:30] Alex: Give her a chef hat!
+[03/01/25, 17:04:40] Jamie: She’d wear it proudly.
+[03/01/25, 18:00:00] Alex: How’s the jobless vibe now?
+[03/01/25, 18:01:10] Jamie: Still weird. Less panicked though.
+[03/01/25, 18:02:20] Alex: Progress! You’re tougher than this.
+[03/01/25, 18:03:30] Jamie: Thanks, Alex. Means a lot.
+[03/01/25, 19:00:00] Jamie: Watched a funny cat video.
+[03/01/25, 19:01:10] Alex: Send it over!
+[03/01/25, 19:02:20] Jamie: "Cat vs. cucumber—total freakout!"
+[03/01/25, 19:03:30] Alex: Dying laughing over here!
+[03/01/25, 19:04:40] Jamie: Glad I could help!
+[03/01/25, 20:00:00] Alex: Movie time soon?
+[03/01/25, 20:01:10] Jamie: Yup, couch is ready.
+[03/01/25, 20:02:20] Alex: Bringing extra popcorn!
+[03/01/25, 20:03:30] Jamie: "Last time we spilled everywhere!"
+[03/01/25, 20:04:40] Alex: Cat probably ate it all.
+[03/01/25, 21:00:00] Jamie: Movie was hilarious.
+[03/01/25, 21:01:10] Alex: That plot twist though!
+[03/01/25, 21:02:20] Jamie: Didn’t see it coming.
+[03/01/25, 21:03:30] Alex: Next time, horror?
+[03/01/25, 21:04:40] Jamie: Sure, if I don’t scream!
+[03/01/25, 22:00:00] Jamie: Bedtime. Exhausted today.
+[03/01/25, 22:01:10] Alex: Sleep well, jobless warrior!
+[03/01/25, 22:02:20] Jamie: Haha, thanks. Night!
+[03/01/25, 22:03:30] Alex: Night! Dream of tacos.
+[03/02/25, 08:00:00] Jamie: Up early. Job hunt again.
+[03/02/25, 08:01:10] Alex: You’re unstoppable!
+[03/02/25, 08:02:20] Jamie: Trying to be. Coffee’s brewing.
+[03/02/25, 08:03:30] Alex: Fixed the machine?
+[03/02/25, 08:04:40] Jamie: Yup, "hammered it into submission!"
+[03/02/25, 08:05:50] Alex: Coffee machine fears you now!
+[03/02/25, 09:00:00] Jamie: Resume sent to five places.
+[03/02/25, 09:01:10] Alex: Awesome! They’ll beg for you.
+[03/02/25, 09:02:20] Jamie: Hope so. Fingers crossed.
+[03/02/25, 10:00:00] Alex: Run today? Clear your head?
+[03/02/25, 10:01:10] Jamie: Maybe. Feeling a bit lighter.
+[03/02/25, 10:02:20] Alex: That’s the spirit!
+[03/02/25, 11:00:00] Jamie: Cat’s napping on my laptop.
+[03/02/25, 11:01:10] Alex: "She’s the real boss now!"
+[03/02/25, 11:02:20] Jamie: Always has been!
+[03/02/25, 11:03:30] Alex: Pet her for luck!