From a25e103189b6ee9eb75199db31bab40be1dc9152 Mon Sep 17 00:00:00 2001
From: priyanshm07 <priyanshmaheswari94@gmail.com>
Date: Tue, 1 Apr 2025 23:02:55 +0530
Subject: [PATCH] update : multi-agent with file ss upload

---
 .../__pycache__/agents.cpython-39.pyc         | Bin 0 -> 4122 bytes
 .../__pycache__/image_input.cpython-39.pyc    | Bin 0 -> 2694 bytes
 .../ai_breakup_recovery_agent/agents.py       |  32 ++++--
 .../ai_breakup_recovery_agent/app.py          |  91 +++++++++++-------
 .../ai_breakup_recovery_agent/image_input.py  |  86 +++++++++++++++++
 .../requirements.txt                          |   2 +-
 6 files changed, 167 insertions(+), 44 deletions(-)
 create mode 100644 ai_agent_tutorials/ai_breakup_recovery_agent/__pycache__/agents.cpython-39.pyc
 create mode 100644 ai_agent_tutorials/ai_breakup_recovery_agent/__pycache__/image_input.cpython-39.pyc
 create mode 100644 ai_agent_tutorials/ai_breakup_recovery_agent/image_input.py

diff --git a/ai_agent_tutorials/ai_breakup_recovery_agent/__pycache__/agents.cpython-39.pyc b/ai_agent_tutorials/ai_breakup_recovery_agent/__pycache__/agents.cpython-39.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..dbe72c28d139df508eaacc8b7ed095159a2552e0
GIT binary patch
literal 4122
zcmcInTW{pH6`uQyZ|hCEc6v!$0xb~O7|ZD;1#)SyyS~MZ<1DgHfcRy=5jh&IB~c|Q
z&st9f(${_K4`}nYZ~Z-bY|)1TMce0~DEb{r8jm+FiUg_0LvbYX@Er0x-}%JX)|MOi
z+r0O;;E$V)#-HiO;$I6t?&E8oq7e<zj2dzCyxF9FD{95<^S0}^qfXpC@49{`>c#!@
zzUz0RLA-Rn<odm6`Ft7kdeO@His*~MZyV>UVo5CHx+Ye{Dz10Lnz)1Ox>y%?alI>g
zj~iPr{S~V>wwguvK_C;0PUlSW7}wsBjKd^EZ+$=WFX(%u&XkG_jXsWfAWwaoy!b~U
z@8N4+M`IC`Xax<?{=8|MqGMZ6+Q09J?jM?>_pD{RpEZIeoTbm;*#OQgo#yLjQ>l3x
z8q1tR`6r*MjBza)XLkN*HdXABN1@<W8a9zK3X{Mvo(LurUu7E38K&hVlD-X9!e*hJ
zs?4${nFwOU>1}oLw@$E%Oqf+<nPo=m*A|S?6ei=8+v%0F>?`~*2Ijo{mk-Rv-2SCx
zzs{tgg$!An$|xn1F_TZzNWpjf2sMl;H^$3<^5JA6HS<*pznQTbaReb;uKgTKdLTj*
zhgk7adik55RPolvo~BZVh!(Rb?o_MGP{?`_9x;=psnXW_;%?wha_6Vs>pPQ(n`v?9
z=-`9n)8o;DN5`Z04?ca~Dq5i^`q=l;g}gd@J}5c~k7d!(Dw4%&n83x%?W8HXu@W*W
zx|X0&yvl_b(J|OCmLsc1H^#1;shZ)gibg(<FnkbmeIe8=dEV2M0J?*wXem?lWpWv6
zl@#qjTHVF%=MDWG8bZw%m%AV1e9UgD!z-ScY0Rh6OlMq&yZh2ySe5P`_)~1bSf22)
z+2u27R4jL*DBj^|iW}hwph)b<W>)EtNAA{GOMa21BQ1S(DfLx3X4j20UV2q*`WRsm
zS&hAN1mWC#7fqwta-Uw0uFL=Kbb95Tf9mhy={KJsYkQG088B6*$9+nVnGWsZ$@|ri
zSQ_Ipp~g4c#(`v!V2QHZ%*aGgBD}NbVqAdE!bENf#Tu5F=CdK2ssvD<=bV?lhcM5O
zKl3@&5~m-Yv9Y8@Ec?@Men+rH9>H;yL|1I0=J^1qY8gpxfb!ErrDH5f4kKSvq^yre
z8Z%rzawHcQQoqHvw?8POWBZy<tYNefk6m=hYnDAmY}?!U&kqu}S!B<`a~2x1pzTCP
zcF1^!2Z;@Rbdf`4Y@Y5TBt4*JiGtIs)ZL(F4#=`c2q^AV5UF;(1jrBZ<bMI=RgC$j
z0g}I29wvwgA%BED3=#FCh$jgM$6==2$r|xYmt%~VZ`?XO|G0t&$1#C}z;iwmJj9Vy
zQ|FvV8AqF35gjrfW6I28I3qZ~S(S0#RyTj+aTb85E#2kt(3j!`!|u#+UJ-Hdl#fFT
zH&2ceKgvLLL^%#fMn;pJYqlRUcOsxo!cZiwDUUauUYQU)#5kzzGPEFEFF&lEFv<rj
zurPt2Gyr5iWKnn_T?p7d+yf^&mnOu#!Z3h%!eeZOm%m?oRl5uF2CvtKOy*A+2l&)@
zjCF^!mjd>NWFwI?O0t^Iy?pO0q`|&BijOJ+tB-tp`)i`OI@1!xZ_H8Ls%WnN0?qwH
zfB*l1;{8PwKh&AU$-L*7=V}qeiP}kUP7?n^R7+sZVq;B8_i(oNZkw53Jd{8onFKs5
zI-~4Y<0o{+Kn))QH*7Wq5zUCcW~fVnV~81WNQp&a0mmkR0TlcgwF9)~u~UVr8*shj
zD^6%3l)rsMCLMQ{9+qIdUJdd^jJbbNElcL>B*}ku00qHA2$*VNFeNM_zG;{Odtro}
zagcO0Idobg45k+RisC2;W=19CN`8n6%EnQafcPc=W`;dQ)&&_##Ft$sT+Z{D`56|h
z?CjS_FKt7#EN)80{?(QYk=X_=LV{6TCd8`bzL&rK6(D}|7{4~O%d@U({K=P~J$M1y
zf7d^_4caTXxkr=VMN>*#h}(80ZaWaS-BR4z-e(Ql$9Djs+Yvno;r{1MF>s%yXG;*g
z%cuFqLu?}?2%Bdx>F;9MS4x8*pgmql<op*4iWozk+$3hI*|%ogiB|C<;r8>R!%!P4
zdl0X59A~wtsd8#TyOs$5Z~FB<W+vBmZb%YJ2*<z35=u?zhf?<fZnZ}6@}pDvbitEb
zD-T&hI#nfN&Z~>12i4#Zn}kfh_U0y9u0snb4=a+mneTN5x@r(ZqJ}w00F;25Lgf?o
zuW5xUgu=dpu8~<Nh&k|G?fDf8GKg?5f3ToC=7i+bPo>J?%W(d#;_3jc2l4EZH;cUd
zXOynBT&6MyUF0edx}_*8iwGY_wN~=ZwwCl7U|DoXr0cuX?Z+}6!`EU(NaO1;rTo+D
zc%)yZ=bOmb*{B)>AzZPizK4e}*QSA_(luUU8~PP$zD*6uTTK$iRREVOx5}e$m0%dv
zQriNv3e(Jw(Nyx;RTqfoZl~!!|7xMm^p;%B`HQB3TQ3#MJV=xWvP!IbcPQ?-8)VO;
z#<^O&>~6Yu6ypU-1(7UYGB(Q5^w#1Dyv))!Q0`5mXg@kW(LF4K*AQ;5l&{NZZ;O$<
jxi!$iJ6JV~<hxCbFtl5%&H6KF4%&l_!RFwV!H@m}SG4Fb

literal 0
HcmV?d00001

diff --git a/ai_agent_tutorials/ai_breakup_recovery_agent/__pycache__/image_input.cpython-39.pyc b/ai_agent_tutorials/ai_breakup_recovery_agent/__pycache__/image_input.cpython-39.pyc
new file mode 100644
index 0000000000000000000000000000000000000000..67969857ed7a717df70107f77a55ff6a7dc48498
GIT binary patch
literal 2694
zcmZuzU2hz<8XnJACfg+a2u%?Z!h8u9C7V(TNYMe2wmDG_r9`x->LRi7j=i%^X2v7i
zv+2%BxCyuU1qi9-2Ju(U)w%ZVxk5tx03^_P?a8KJ>}qV!Yy0*4^*kQ3xY+UF_w%d2
zWt$70_domc^v}+I29NqLG}NOg_X^ZU$d3KoFM@vH#z7tyQ9p9yFprD5{@m%B1lG*E
zag=xZ9oQG==Zb~?!l`Vrzewk3^1atzqVu!^{dszhE<k^QF7_`rZ?PqI@fh_lvE{v&
z=n@OqC3^k@Wr$v&7f;~y`(Al1^w{O;e~{N-L0<C;yFxFql^?tx{bR87zW2bRQGex|
zm9o?JSG}tsx$AY8e+93+E;7+(#!77mI|a!Y4B~B8@RGym{H;3Ox6ih?Cq%C9fNbu@
zR5P{nWs0nV(_aX^eQv^|u0o@|N9fRdj2`(EJw``J2glx#e~9+NL!bIIIPsBp=s$`c
zdq=@x0QQ6pzAXsFq@=i#tRj+OZVke<;$?<+cW#^DtL+;m{MYW*wzYC3WI=Q{HVMo*
z+f67cR+_m~loKYkjKL5SDl%av$(Wvyyk_0dbe!G+C7PIk7nw<h^=QN-qh=mn2Se-N
zFP>b6?SK4x{cFXfTCXIZl2VNeGG=OgKqOzk#niqQ)%sRC785>Xx5-eglLMwi!PfG;
zSR+*h0zLpQN<GlE7Lt?PiH4Gq{kj@RmI|<S+OAn|E#qZX>s~cAog6|ka6(&+$`x#I
zA&AiuN_=@4WGOt){N~}!rC}8Y8p@XXnMZ@4kVZ7r{$B7Xq|swU;}c&;#~u_Dl;BAW
zDVT3Qz6oZ5^O#^&u@sEu4TGFm`!U2DvP1G|WAl-YiN+Ptn#oc%%Qx-orkzz>;Z3`t
zDIbU$+dTfAHf=Yw>fk)5O{J1C2(i%09VS^$G#4P@)!9+5KvGc@b;(mVjk!`aQ@w7|
zeC#Y}7}jI<yiTXm%10v4#ewx<X8uO=&HFuk^BJYPr><?_9SY9y(G)|jYANkTP*X6q
z3KCU;eQ+Lk@3wf`;!~|@BRHwqzShILL@7RDYnxdHn-qAp0rf8SomHwH>YU-RP?|#z
zaV|*7tJ=ChVK^scRznV&FF)wvFV1GHiZk0XqH>;%HCVTbvzq66O-o)??j00@Ra%1M
z*7_kElL;46HMg#VLjDj^e3xgCJhg#uG0jqqc?mlpmD*+=d{c0~vX<3lnlmC{ft|A1
zX+GPPQ2*5CnKDhl1q<WTQaIpx+!oV_r<gq?MU{i5krc&QmYPrg>9FhsqSjFT*qxG3
zal0Q(vPZdSF_MmFsl$3&vF7#-2-<MZE`o5`B#cB_gSjP$0lgD4gs{%CjU|EnNjQv@
z5a@jy639+nRINJ~GDm-F*LqjiMmbSqv$VbS#m=3b!RGGH;Pb5qUElbenmJPO!9JVb
z0}2XDQZN%jgqd03rP8u)EjCdhD9cTxEq|J%Aab9IgR&b-AMhzHX&PTBGsnsam!dQQ
zFqMq#oaLXVnwRX-R{$e`(18WaU>4+-JYI!)wE~Uj`^f&|j%_=Cg<*{R-{l)H^9*U*
zLoA~>wxBR*R<(EN9b39aj;{SZpl=-<0*AP32WZ>U^<H@2y9<1;qv^NEI|}W^1KbiH
zhR4WWJ@>uXT|7tOci~;QZft*c@x;xCGnA4SFxqmxMd=It)976&Jiwas+F=}+<~ibU
z1@bIA%+kIJ1<+!UoMCYFDWHjR)s;@+@Ss{e^nQU0&teB^-QGElidFOamXt5D;S>Ja
zz<0WHj-FlAO(5%11`uKS7Bpsl>tV`VwV41)r5nlfR(!5ye|P<u$dRr|W)x!(d%hN~
zoJOK8h9{?A2W84u3lxH%1aT0$GKizZUqW&X<juL$`=_^8WzDDeSe^sOF&P5*02dHf
zXNrqQW-$emHYq?zOle%Q1GpmGVy=7gP0)C+>&v&HH!(QGNYT99hNzMvwR{H{b1FA*
z_hr)|St)wXL5G4<LBS{|E==Z<Tj8LidYKSe&RVEkb`uUvJqjNt_Q{09iwWR6K-z*f
z@em5-qw4@u;XVT7Yao4+4;O>kX8_d4*u`JIXSXiFMGu$tEbQ_fyVG8!CUP@RCBfm-
rp)&agm|ExTQ~OP%-i2lX&Hf3D5<gi^t|XU3mk^f@<p0@;meBtJRuwL?

literal 0
HcmV?d00001

diff --git a/ai_agent_tutorials/ai_breakup_recovery_agent/agents.py b/ai_agent_tutorials/ai_breakup_recovery_agent/agents.py
index 757683c..f9ff6e8 100644
--- a/ai_agent_tutorials/ai_breakup_recovery_agent/agents.py
+++ b/ai_agent_tutorials/ai_breakup_recovery_agent/agents.py
@@ -1,9 +1,10 @@
 from agno.agent import Agent
 from agno.team.team import Team
-from agno.models.openai import OpenAIChat
+from agno.models.google import Gemini
 from agno.tools.duckduckgo import DuckDuckGoTools
 from PIL import Image
 import easyocr
+import os
 
 # --- Therapist Agent ---
 def create_therapist_agent():
@@ -16,7 +17,10 @@ def create_therapist_agent():
             "Offer coping strategies without being dismissive.",
             "Validate their experiences and provide emotional support.",
         ],
-        model=OpenAIChat(id="gpt-4o-mini"),
+        model=Gemini(
+            id="gemini-2.0-flash",
+            api_key=os.environ.get("GEMINI_API_KEY")
+        ),
         tools=[DuckDuckGoTools()],
         add_datetime_to_instructions=True,     # Add timestamp context
         show_tool_calls=True,
@@ -31,10 +35,13 @@ def create_closure_agent():
         instructions=[
             "Create emotional messages that express raw, honest feelings.",
             "The messages should NOT be sent — they are only for emotional release.",
-            "Format the output clearly with a header: **Message Drafts You Shouldn’t Send**",
+            "Format the output clearly with a header: **Message Drafts You Shouldn't Send**",
             "Ensure the tone is heartfelt, authentic, and honest.",
         ],
-        model=OpenAIChat(id="gpt-4o-mini"),
+        model=Gemini(
+            id="gemini-2.0-flash",
+            api_key=os.environ.get("GEMINI_API_KEY")
+        ),
         tools=[DuckDuckGoTools()],
         add_datetime_to_instructions=True,    # Add timestamp context
         show_tool_calls=True,
@@ -55,7 +62,10 @@ def create_routine_agent():
             "Include healthy distractions, such as hobbies, reading, or new experiences.",
             "Format the output clearly with a header: **Daily Recovery Routine**",
         ],
-        model=OpenAIChat(id="gpt-4o-mini"),
+        model=Gemini(
+            id="gemini-2.0-flash",
+            api_key=os.environ.get("GEMINI_API_KEY")
+        ),
         tools=[DuckDuckGoTools()],
         add_datetime_to_instructions=True,
         show_tool_calls=True,
@@ -75,7 +85,11 @@ def create_honesty_agent():
             "Include reasons why the user should move on, based on the situation.",
             "Format the output clearly with a header: **Brutal Honesty**",
         ],
-        model=OpenAIChat(id="gpt-4o-mini"),
+        # model=Gemini(id="gemini-2.0-flash"),
+        model=Gemini(
+            id="gemini-2.0-flash",
+            api_key=os.environ.get("GEMINI_API_KEY")
+        ),
         tools=[DuckDuckGoTools()],
         add_datetime_to_instructions=True,
         show_tool_calls=True,
@@ -98,7 +112,11 @@ def create_breakup_team():
     return Team(
         name="Breakup Recovery Team",
         mode="coordinate",  # Team execution mode: coordinate or parallel
-        model=OpenAIChat(id="gpt-4o-mini"),
+        # model=Gemini(id="gemini-2.0-flash"),
+        model=Gemini(
+            id="gemini-2.0-flash",
+            api_key=os.environ.get("GEMINI_API_KEY")
+        ),
         members=[
             create_therapist_agent(),
             create_closure_agent(),
diff --git a/ai_agent_tutorials/ai_breakup_recovery_agent/app.py b/ai_agent_tutorials/ai_breakup_recovery_agent/app.py
index b11054b..f36f852 100644
--- a/ai_agent_tutorials/ai_breakup_recovery_agent/app.py
+++ b/ai_agent_tutorials/ai_breakup_recovery_agent/app.py
@@ -1,5 +1,6 @@
 import streamlit as st
 from agents import create_breakup_team
+from image_input import analyze_chat_screenshot
 from PIL import Image
 
 # --- Streamlit Layout ---
@@ -7,16 +8,16 @@ st.title("💔 Breakup Recovery Agent Team")
 st.write("Receive support and emotional outlet messages from specialized AI agents.")
 
 # --- API Keys Input ---
-openai_api_key = st.sidebar.text_input("OpenAI API Key", type="password")
+gemini_api_key = st.sidebar.text_input("Gemini API Key", type="password")
 submit_button = st.sidebar.button("Submit")
 
 # --- Store API Keys ---
 if submit_button:
-    if openai_api_key:
-        st.session_state["openai_api_key"] = openai_api_key
-        st.success("API keys saved successfully!")
+    if gemini_api_key:
+        st.session_state["gemini_api_key"] = gemini_api_key
+        st.success("API key saved successfully!")
     else:
-        st.error("Please enter both API keys!")
+        st.error("Please enter the API key!")
 
 # --- User Input ---
 user_input = st.text_area("Describe how you're feeling...")
@@ -24,47 +25,65 @@ user_input = st.text_area("Describe how you're feeling...")
 # --- Image Upload ---
 uploaded_image = st.file_uploader("Upload a chat screenshot (optional)", type=["png", "jpg", "jpeg"])
 
-# --- Display Uploaded Image ---
-extracted_text = ""
+# --- Display Uploaded Image and Analysis ---
 if uploaded_image:
     image = Image.open(uploaded_image)
     st.image(image, caption="Uploaded Screenshot", use_column_width=True)
+    
+    if st.button("Analyze Chat"):
+        with st.spinner("Analyzing chat patterns..."):
+            if not st.session_state.get("gemini_api_key"):
+                st.error("Please enter your Gemini API key in the sidebar first.")
+            else:
+                try:
+                    # Set environment variable for the analysis
+                    import os
+                    os.environ["GEMINI_API_KEY"] = st.session_state["gemini_api_key"]
+                    
+                    # Analyze the chat screenshot
+                    analysis = analyze_chat_screenshot(uploaded_image)
+                    
+                    # Display the analysis
+                    st.subheader("🔍 Chat Analysis")
+                    st.markdown(analysis)
+                except Exception as e:
+                    st.error(f"An error occurred during analysis: {str(e)}")
+                    st.info("Please check your Gemini API key and try again.")
 
-    # # ✅ Extract text from the image
-    # extracted_text = extract_text_with_easyocr(image)
-    # st.text_area("Extracted Text from Image:", extracted_text)
-
-# --- Execute Agent Team ---
+# --- Execute Agent Team (only for text input) ---
 if st.button("Get Recovery Support"):
     with st.spinner("Agents are processing..."):
-        if not user_input and not uploaded_image:
+        if not user_input:
             st.error("Please enter what you feel.")
+        elif not st.session_state.get("gemini_api_key"):
+            st.error("Please enter your Gemini API key in the sidebar first.")
         else:
-            breakup_team = create_breakup_team()
+            try:
+                breakup_team = create_breakup_team()
 
-            # Set environment variables
-            import os
-            os.environ["OPENAI_API_KEY"] = st.session_state.get("openai_api_key", "")
+                # Set environment variables
+                import os
+                os.environ["GEMINI_API_KEY"] = st.session_state["gemini_api_key"]
 
-            # Prepare the input
-            input_data = user_input
-            if uploaded_image:
-                input_data += "\n\n[Attached Image Included]"  # Add image indicator
+                # Execute the team with text input
+                response = breakup_team.run(user_input)
 
-            # Execute the team
-            response = breakup_team.run(user_input)
+                # Display responses
+                st.subheader("💡 Team's Responses")
 
-            # Display responses
-            st.subheader("💡 Team's Responses")
+                # Display individual agent responses
+                if response.member_responses:
+                    for index, member_response in enumerate(response.member_responses, start=0):
+                        st.markdown(f"### 🛡️ {response.tools[index]['tool_args']['agent_name']}")
+                        st.markdown(member_response.content)
+                else:
+                    st.warning("⚠️ No individual agent responses received. This might be due to an API error or configuration issue.")
+                    st.info("Please check your Gemini API key and try again.")
 
-            # Display individual agent responses
-            if response.member_responses:
-                for member_response in response.member_responses:
-                    st.markdown(f"### 🛡️ {member_response.agent_id}")
-                    st.write(member_response.content)
-            else:
-                st.warning("⚠️ No individual agent responses received. Showing team leader's summary only.")
-
-            # Display team leader's final summary
-            st.subheader("📜 Team Leader's Summary")
-            st.write(response.content)
+                # Display team leader's final summary
+                if response.content:
+                    st.subheader("📜 Team Leader's Summary")
+                    st.markdown(response.content)
+            except Exception as e:
+                st.error(f"An error occurred: {str(e)}")
+                st.info("Please check your Gemini API key and try again.")
diff --git a/ai_agent_tutorials/ai_breakup_recovery_agent/image_input.py b/ai_agent_tutorials/ai_breakup_recovery_agent/image_input.py
new file mode 100644
index 0000000..05908be
--- /dev/null
+++ b/ai_agent_tutorials/ai_breakup_recovery_agent/image_input.py
@@ -0,0 +1,86 @@
+from agno.agent import Agent
+from agno.media import Image
+from agno.models.google import Gemini
+from agno.tools.duckduckgo import DuckDuckGoTools
+from PIL import Image as PILImage
+import base64
+from io import BytesIO
+import os
+
+def load_image(image_data):
+    """Load and prepare image using PIL"""
+    # Convert image data to PIL Image
+    img = PILImage.open(image_data).convert('RGB')
+    
+    # Convert PIL image to bytes
+    buffered = BytesIO()
+    img.save(buffered, format="JPEG")
+    img_bytes = buffered.getvalue()
+    
+    return img_bytes
+
+# # Check for Google API key
+# if "GOOGLE_API_KEY" not in os.environ:
+#     raise EnvironmentError(
+#         "Please set your GOOGLE_API_KEY environment variable. "
+#         "You can get one from https://makersuite.google.com/app/apikey"
+#     )
+
+def create_chat_analysis_agent():
+    """Creates a specialized agent for analyzing chat patterns"""
+    return Agent(
+        name="Chat Analysis Agent",
+        role="You are a specialized agent that analyzes chat screenshots for relationship patterns and communication issues.",
+        instructions=[
+            "Analyze the chat screenshot for the following patterns:",
+            "1. Communication Patterns: Identify recurring themes, response times, and conversation flow",
+            "2. Passive-Aggression: Look for subtle hostile or manipulative language",
+            "3. Manipulation: Identify gaslighting, guilt-tripping, or controlling behaviors",
+            "4. Mixed Signals: Detect inconsistent messages or unclear intentions",
+            "Provide a detailed analysis with specific examples from the chat",
+            "Format the output clearly with sections for each pattern type",
+            "Be objective and focus on observable patterns rather than making assumptions",
+        ],
+        model=Gemini(
+            id="gemini-2.0-flash",
+            api_key=os.environ.get("GEMINI_API_KEY")
+        ),
+        tools=[DuckDuckGoTools()],
+        markdown=True,
+    )
+
+def analyze_chat_screenshot(image_data):
+    """Analyze a chat screenshot using the specialized agent"""
+    try:
+        # Load and process the image
+        image_bytes = load_image(image_data)
+        
+        # Create the chat analysis agent
+        agent = create_chat_analysis_agent()
+        
+        # Analyze the chat screenshot
+        response = agent.run(
+            "Please analyze this chat screenshot for communication patterns, passive-aggression, manipulation, and mixed signals.",
+            images=[Image(content=image_bytes)]
+        )
+        
+        return response.content
+        
+    except Exception as e:
+        return f"Error analyzing chat screenshot: {str(e)}"
+
+# Example local image path - replace with your image path
+image_path = "breakup.jpg"
+
+try:
+    # Load and process the image
+    image_data = load_image(image_path)
+    
+    # Create the response using the agent
+    agent.print_response(
+        "Tell me about this image and give me the latest news about it.",
+        images=[Image(content=image_data)],
+        stream=True,
+    )
+except Exception as e:
+    print(f"Error processing image: {e}")
\ No newline at end of file
diff --git a/ai_agent_tutorials/ai_breakup_recovery_agent/requirements.txt b/ai_agent_tutorials/ai_breakup_recovery_agent/requirements.txt
index 25c0952..84af6db 100644
--- a/ai_agent_tutorials/ai_breakup_recovery_agent/requirements.txt
+++ b/ai_agent_tutorials/ai_breakup_recovery_agent/requirements.txt
@@ -1,3 +1,3 @@
-agno==1.2.4
+agno>=1.2.4
 pillow==11.1.0
 streamlit==1.44.0