add location

1 files changed, 39 insertions, 4 deletions

diff --git a/main.py b/main.py
index e760fd5..52c5888 100644
--- a/main.py
+++ b/main.py
@@ -137,7 +137,22 @@ async def guest_middleware(handler, event, data):
                 replied_text = None
                 
             prompt = ""
-            
+
+            if replied and replied.location:
+                loc = replied.location
+                prompt += (
+                    f"<ответ на локацию>\n"
+                    f"{loc.latitude}, {loc.longitude}\n"
+                    f"</ответ на локацию>\n"
+                )
+        
+            if message.location:
+                loc = message.location
+                prompt += (
+                    f"<локация>\n"
+                    f"{loc.latitude}, {loc.longitude}\n"
+                    f"</локация>\n"
+                )
             if replied and replied.poll:
                 poll = replied.poll
                 if poll.allows_multiple_answers:
@@ -222,7 +237,7 @@ async def guest_middleware(handler, event, data):
     return await handler(event, data)
 
     
-@router.message(F.content_type.in_({"text", "photo", "document", "poll"}))
+@router.message(F.content_type.in_({"text", "photo", "document", "poll", "location"}))
 async def text_handler(message: Message):
     chat_type = message.sender_chat.type if message.sender_chat else ""
     user_id = message.sender_chat.id if message.sender_chat else message.from_user.id
@@ -327,9 +342,28 @@ async def text_handler(message: Message):
             print(f"{RED} -- {e}{RESET}")
             return
 
+    if replied and replied.location:
+        loc = replied.location
+    
+        prompt += (
+            f"<ответ на локацию>\n"
+            f"{loc.latitude}, {loc.longitude}\n"
+            f"</ответ на локацию>\n"
+        )
+
+    if message.location:
+        loc = message.location
+    
+        prompt += (
+            f"<локация>\n"
+            f"{loc.latitude}, {loc.longitude}\n"
+            f"</локация>\n"
+        )
+    
     elif replied and replied.photo and IMAGE_MODEL != "":
         try:
-            msg = await message.reply("👀")
+            if not msg:
+                msg = await message.reply("👀")
         except Exception:
             return
         try:
@@ -345,7 +379,8 @@ async def text_handler(message: Message):
             prompt = f"<ответ на фото>{image_text}</ответ на фото>\n<подпись в ответе>{replied.caption}</подпись в ответе>\n"
         else:
             prompt = f"<ответ на фото>{image_text}</ответ на фото>\n<подпись в ответе>{replied.caption}\n"
-        await msg.edit_text("💬")
+        if not msg:
+            await msg.edit_text("💬")
 
     elif replied and replied.text:
         replied_text = message.reply_to_message.text