Update main.py
Browse files
    	
        main.py
    CHANGED
    
    | 
         @@ -125,7 +125,7 @@ def get_image_url(message_id): 
     | 
|
| 125 | 
         
             
                    print(f"Error getting image content: {e}")
         
     | 
| 126 | 
         
             
                    return None
         
     | 
| 127 | 
         | 
| 128 | 
         
            -
            @ 
     | 
| 129 | 
         
             
            def handle_image_message(event):
         
     | 
| 130 | 
         
             
                image_url = get_image_url(event.message.id)
         
     | 
| 131 | 
         
             
                if not image_url:
         
     | 
| 
         @@ -141,7 +141,7 @@ def handle_image_message(event): 
     | 
|
| 141 | 
         
             
                user_id = event.source.user_id
         
     | 
| 142 | 
         
             
                user_sessions[user_id] = {"image_url": image_url}
         
     | 
| 143 | 
         | 
| 144 | 
         
            -
            @ 
     | 
| 145 | 
         
             
            def handle_text_message(event):
         
     | 
| 146 | 
         
             
                user_id = event.source.user_id
         
     | 
| 147 | 
         
             
                if user_id in user_sessions and "image_url" in user_sessions[user_id]:
         
     | 
| 
         | 
|
| 125 | 
         
             
                    print(f"Error getting image content: {e}")
         
     | 
| 126 | 
         
             
                    return None
         
     | 
| 127 | 
         | 
| 128 | 
         
            +
            @line_handler.add(MessageEvent, message=ImageMessage)
         
     | 
| 129 | 
         
             
            def handle_image_message(event):
         
     | 
| 130 | 
         
             
                image_url = get_image_url(event.message.id)
         
     | 
| 131 | 
         
             
                if not image_url:
         
     | 
| 
         | 
|
| 141 | 
         
             
                user_id = event.source.user_id
         
     | 
| 142 | 
         
             
                user_sessions[user_id] = {"image_url": image_url}
         
     | 
| 143 | 
         | 
| 144 | 
         
            +
            @line_handler.add(MessageEvent, message=TextMessage)
         
     | 
| 145 | 
         
             
            def handle_text_message(event):
         
     | 
| 146 | 
         
             
                user_id = event.source.user_id
         
     | 
| 147 | 
         
             
                if user_id in user_sessions and "image_url" in user_sessions[user_id]:
         
     |