Upload 6 files

src/extract_features.py

@@ -0,0 +1,72 @@
+import re
+import random
+from collections import Counter, defaultdict
+
+def parse_chat(file_path):
+    pattern = r"(\d{1,2}/\d{1,2}/\d{2,4}), (\d{1,2}:\d{2}) - ([^:]+): (.*)"
+    messages = []
+
+    with open(file_path, "r", encoding="utf-8") as f:
+        for line in f:
+            match = re.match(pattern, line)
+            if match:
+                date, time, sender, text = match.groups()
+                
+                # Normalize names
+                if sender == "ak":
+                    sender = "Aman"
+                elif sender == "Sarah con H":
+                    sender = "Sarah"
+                
+                messages.append({
+                    "date": date,
+                    "time": time,
+                    "sender": sender,
+                    "text": text.strip()
+                })
+    return messages
+
+
+def extract_inside_jokes(messages):
+    funny_candidates = []
+    cute_candidates = []
+    memory_candidates = []
+    phrase_counter = Counter()
+
+    funny_keywords = ["lol", "😂", "🤣", "lmao", "funny", "haha", "hehe","hahaha"]
+    cute_keywords = ["miss", "thank", "sweet", "cute", "proud", "happy","aww","glad"]
+
+    for msg in messages:
+        text = msg["text"].lower()
+
+        # Funny moments
+        if any(k in text for k in funny_keywords):
+            funny_candidates.append(msg["text"])
+
+        # Cute/emotional moments
+        if any(k in text for k in cute_keywords):
+            cute_candidates.append(msg["text"])
+
+        # Memorable random moments
+        if len(msg["text"].split()) > 4:  # skip too short
+            memory_candidates.append(msg["text"])
+
+        # Count repeated words
+        phrase_counter.update(text.split())
+
+    top_words = [w for w, c in phrase_counter.most_common(40)]
+
+    return {
+        "funny": funny_candidates,
+        "cute": cute_candidates,
+        "memories": memory_candidates,
+        "top_words": top_words
+    }
+
+
+def random_memory(messages):
+    """Returns a random meaningful moment."""
+    long_messages = [m["text"] for m in messages if len(m["text"]) > 10]
+    if not long_messages:
+        return "One of your old conversations ❤️"
+    return random.choice(long_messages)

src/generate_jokes.py

@@ -0,0 +1,153 @@
+import json
+import random
+from openai import OpenAI
+import os
+from dotenv import load_dotenv
+from extract_features import parse_chat, extract_inside_jokes, random_memory
+
+# Load environment variables
+load_dotenv()
+
+def generate_inside_jokes(count=3):
+    # Configure Groq API
+    client = OpenAI(
+        api_key=os.environ.get("GROQ_API_KEY"),
+        base_url="https://api.groq.com/openai/v1",
+    )
+
+    messages = parse_chat("chat.txt")
+    data = extract_inside_jokes(messages)
+
+    # meaningful memory for Sarah
+    random_real_memory = select_meaningful_memory(messages, client)
+
+    prompt = f"""
+You are a comedian whoose task is to generate funny inside jokes for two best friends: Aman and Sarah.
+They have a very close, playful, supportive and affectionate friendship.
+
+Here is the context from their WhatsApp chats:
+
+Funny lines they've said:
+{json.dumps(data['funny'][:30], indent=2)}
+
+Cute moments between them:
+{json.dumps(data['cute'][:20], indent=2)}
+
+Shared vocabulary & patterns:
+{json.dumps(data['top_words'][:20], indent=2)}
+
+One real memory to set the mood:
+"{random_real_memory}"
+
+Your Goal:
+Generate {count} ORIGINAL inside jokes that sound exactly like something they would say to each other.
+The jokes should be a mix of funny, teasing, and heartfelt(more of funny).
+
+Guidelines:
+1. **Be Human**: Use casual language, slang (if fits their vibe), and natural phrasing. Avoid robotic or stiff sentence structures.
+2. **Show, Don't Tell**: Instead of saying "You are funny," say "Remember that time you tried to tell a joke and choked on water? Classic."
+3. **Mix Humor and Heart**: The best inside jokes often come from a place of love. "Fries before guys" is funny but also shows loyalty.
+4. **Use Context**: Reference their shared vocabulary or specific funny/cute moments provided above.
+5. **Be creative**: The best inside jokes often come from a place of love. "Fries before guys" is funny but also shows loyalty.
+6. Try to make atleast one line jokes
+
+
+Output Format:
+
+INSIDE JOKES:
+1. [Joke/Phrase] 
+2. [Joke/Phrase] 
+...
+
+MEMORY FOR SARAH:
+"{random_real_memory}"
+    """
+
+    response = client.chat.completions.create(
+        model="llama-3.3-70b-versatile",
+        messages=[
+            {"role": "user", "content": prompt}
+        ]
+    )
+
+    return response.choices[0].message.content
+
+
+def select_meaningful_memory(messages, client):
+    """Selects a meaningful memory using LLM."""
+    # Filter for potentially interesting messages to save tokens
+    candidates = [m["text"] for m in messages if len(m["text"]) > 15]
+    if not candidates:
+        return "Remember that time we couldn't stop laughing? Good times."
+    
+    # Sample if too many
+    if len(candidates) > 50:
+        candidates = random.sample(candidates, 50)
+
+    candidates_str = "\n".join([f"- {c}" for c in candidates])
+
+    prompt = f"""
+Here are some excerpts from a chat history between two best friends, Aman and Sarah:
+
+{candidates_str}
+
+Your Task:
+Select ONE specific message from this list that represents a "meaningful memory".
+Criteria for "meaningful":
+- A plan to meet (e.g., "Let's go to that movie", "Dinner tonight?")
+- A sweet/emotional moment
+- An inside joke or funny observation
+- NOT just "Hello" or "Okay"
+
+Return ONLY the exact text of the selected message. Nothing else.
+    """
+
+    try:
+        response = client.chat.completions.create(
+            model="llama-3.3-70b-versatile",
+            messages=[{"role": "user", "content": prompt}]
+        )
+        return response.choices[0].message.content.strip()
+    except Exception:
+        return random.choice(candidates)
+
+
+def generate_motivation():
+    # Configure Groq API
+    client = OpenAI(
+        api_key=os.environ.get("GROQ_API_KEY"),
+        base_url="https://api.groq.com/openai/v1",
+    )
+
+    try:
+        with open("sarah_about.txt", "r", encoding="utf-8") as f:
+            about_sarah = f.read()
+    except FileNotFoundError:
+        return "You are stronger than you know. Keep shining! ✨"
+
+    prompt = f"""
+You are a supportive, wise, and kind friend.
+Read this context about Sarah:
+"{about_sarah}"
+
+Your Task:
+Generate a short, powerful, and personalized motivational thought for Sarah.
+It should directly address her struggles (past burdens, insecurity) but focus on her strengths (kindness, resilience, future in Air Force).
+Make it sound like it's coming from a place of deep understanding and belief in her.
+Keep it under 2 sentences.
+
+Output ONLY the motivational message.
+    """
+
+    response = client.chat.completions.create(
+        model="llama-3.3-70b-versatile",
+        messages=[
+            {"role": "user", "content": prompt}
+        ]
+    )
+
+    return response.choices[0].message.content
+
+
+if __name__ == "__main__":
+    print(generate_inside_jokes(5))

src/requirements.txt

@@ -0,0 +1,3 @@
+openai==2.9.0
+python-dotenv==1.2.1
+streamlit==1.52.1

src/sarah_about.txt

@@ -0,0 +1,14 @@
+Sarah is a kind hearted, mature and a beautiful person. 
+She went through a lot of hardships in her life but she never lost her smile.
+She is a true inspiration to everyone around her in the past.
+Like all her life she thinks that she did some mistakes which cannot be forgiven that's why
+she always felt insecure with her relationship with her boyfriend.
+She has a very pure heart and is very straightforward with her thoughts and actions.
+She wants to join US air force and finish her education, currently she is doing bachelors in
+pshycology.
+Though her boyfriend had also made mistakes but he didn't accept that and puts all the 
+blame on her.
+She is kind of carrying her past burden with her in the present.
+She has a friend Aman who never judges her and will always stand by her in times of need as a friend.
+She is one the most brave and courageous person I know.
+

src/streamlit_app.py

@@ -1,40 +1,133 @@
-import altair as alt
-import numpy as np
-import pandas as pd
-import streamlit as st
-
-"""
-# Welcome to Streamlit!
-
-Edit `/streamlit_app.py` to customize this app to your heart's desire :heart:.
-If you have any questions, checkout our [documentation](https://docs.streamlit.io) and [community
-forums](https://discuss.streamlit.io).
-
-In the meantime, below is an example of what you can do with just a few lines of code:
-"""
-
-num_points = st.slider("Number of points in spiral", 1, 10000, 1100)
-num_turns = st.slider("Number of turns in spiral", 1, 300, 31)
-
-indices = np.linspace(0, 1, num_points)
-theta = 2 * np.pi * num_turns * indices
-radius = indices
-
-x = radius * np.cos(theta)
-y = radius * np.sin(theta)
-
-df = pd.DataFrame({
-    "x": x,
-    "y": y,
-    "idx": indices,
-    "rand": np.random.randn(num_points),
-})
-
-st.altair_chart(alt.Chart(df, height=700, width=700)
-    .mark_point(filled=True)
-    .encode(
-        x=alt.X("x", axis=None),
-        y=alt.Y("y", axis=None),
-        color=alt.Color("idx", legend=None, scale=alt.Scale()),
-        size=alt.Size("rand", legend=None, scale=alt.Scale(range=[1, 150])),
-    ))
+import streamlit as st
+import os
+import random
+from generate_jokes import generate_inside_jokes, generate_motivation
+
+# Page Config
+st.set_page_config(
+    page_title="Memory Forever",
+    page_icon="💛",
+    layout="wide"
+)
+
+# Custom CSS
+st.markdown("""
+    <style>
+    .main {
+        background-color: #f0f2f6;
+    }
+    .stButton>button {
+        width: 100%;
+        background-color: #ff4b4b;
+        color: white;
+        border-radius: 10px;
+        height: 50px;
+        font-size: 20px;
+    }
+    .joke-card {
+        background-color: white;
+        color: black;
+        padding: 20px;
+        border-radius: 10px;
+        box-shadow: 0 4px 6px rgba(0,0,0,0.1);
+        margin-bottom: 15px;
+        border-left: 5px solid #ff4b4b;
+    }
+    .memory-card {
+        background-color: #fff3cd;
+        padding: 15px;
+        border-radius: 10px;
+        border: 1px solid #ffeeba;
+        color: #856404;
+        font-style: italic;
+    }
+    .motivation-card {
+        background-color: #d4edda;
+        color: #155724;
+        padding: 20px;
+        border-radius: 10px;
+        border-left: 5px solid #28a745;
+        margin-top: 20px;
+        font-weight: bold;
+    }
+    h1 {
+        color: #ff4b4b;
+        text-align: center;
+        font-family: 'Helvetica', sans-serif;
+    }
+    </style>
+    """, unsafe_allow_html=True)
+
+# Header
+st.title("💛 Memory Forever 👩‍✈️")
+st.markdown("---")
+
+# Layout: 2 Columns
+col1, col2 = st.columns([1, 2])
+
+with col1:
+    st.header("📸 Past Gallery")
+    
+    # Load images from 'sarah' directory
+    image_dir = "sarah"
+    if os.path.exists(image_dir):
+        images = [f for f in os.listdir(image_dir) if f.lower().endswith(('.png', '.jpg', '.jpeg', '.gif'))]
+        
+        if images:
+            # Display a random image or a grid
+            selected_image = random.choice(images)
+            st.image(os.path.join(image_dir, selected_image), caption="Classic Sarah Moment", use_container_width=True)
+            
+            # Optional: Show more images in an expander
+            with st.expander("See more photos"):
+                for img in images:
+                    st.image(os.path.join(image_dir, img), use_container_width=True)
+        else:
+            st.info("No images found in 'sarah' folder. Add some photos to see them here!")
+    else:
+        st.warning("Folder 'sarah' not found. Please create it and add images.")
+
+    st.markdown("---")
+    st.subheader("⚙️ Settings")
+    count = st.slider("How many jokes?", 1, 10, 3)
+    
+    st.markdown("---")
+    st.subheader("💪 Daily Boost")
+    if st.button("Need a quick motivation ✨"):
+        with st.spinner("Finding the right words..."):
+            motivation = generate_motivation()
+            st.markdown(f'<div class="motivation-card">{motivation}</div>', unsafe_allow_html=True)
+
+with col2:
+    st.header("✨ Memory jokes fusion")
+    
+    if st.button("Generate 🎲"):
+        with st.spinner("Cooking up some humor..."):
+            try:
+                jokes_text = generate_inside_jokes(count)
+                
+                # Parse the output to separate jokes and memory
+                if "MEMORY FOR SARAH:" in jokes_text:
+                    parts = jokes_text.split("MEMORY FOR SARAH:")
+                    jokes_section = parts[0].replace("INSIDE JOKES:", "").strip()
+                    memory_section = parts[1].strip().strip('"')
+                else:
+                    jokes_section = jokes_text
+                    memory_section = "Remember that time..."
+
+                # Display Jokes
+                st.markdown("### 🎭 Inside Jokes")
+                for line in jokes_section.split('\n'):
+                    if line.strip():
+                        st.markdown(f'<div class="joke-card">{line.strip()}</div>', unsafe_allow_html=True)
+
+                # Display Memory
+                st.markdown("### 💌 A Memory to Cherish")
+                st.markdown(f'<div class="memory-card">"{memory_section}"</div>', unsafe_allow_html=True)
+                
+            except Exception as e:
+                st.error(f"Oof! Something went wrong: {e}")
+
+# Footer
+st.markdown("---")
+st.markdown("<div style='text-align: center; color: gray;'>Made with 💖 by Aman</div>", unsafe_allow_html=True)